Windows 10安裝TensorFlow之TensorFlow篇
TensorFlow使用GPU版本效率會更高,本文將講解TensorFlow-GPU版本的安裝,由TensorFlow官網可知安裝GPU版本時所需要的硬件和軟件的requirements,它需要NVIDIA的GPU(硬件)和NVIDIA 驅動、CUDA、CUDNN(軟件),其他可以不用。
接下來將講解TensorFlow的安裝。
版本對應
請看Windows 10安裝TensorFlow之CUDA篇
安裝
- CUDA安裝
請看Windows 10安裝TensorFlow之CUDA篇 - 安裝TensorFlow
-
方法一:
PYPI官網–> 搜索 tensorflow-gpu(一般第一個就是了)–> 點擊Release history–> 點擊你要的版本–> 點擊Download files–> 下載找到你的版本下載的版本
-
方法二:直接安裝
pip install tensorflow-gpu==對應版本
eg: pip install tensorflow-gpu==1.15.0rc3
注:建議安裝Anaconda+Pycharm使用更佳
-