1. 程式人生 > >win10+pycharm+anaconda3+tensorflow-gpu安裝

win10+pycharm+anaconda3+tensorflow-gpu安裝

1.安裝好cuda和cuDNN

2.檢視所匹配版本進行安裝tensorflow-gpu。如下:

tensorflow-gpu v1.9.0 | cuda9.0 |  cuDNN7.1.4可行  | 備註:7.0.4/ 7.0.5/ 7.1.2不明確

tensorflow-gpu v1.8.0 | cuda9.0 |  cuDNN  不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

tensorflow-gpu v1.7.0 | cuda9.0 |  cuDNN  不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

tensorflow-gpu v1.6.0 | cuda9.0 |  cuDNN  不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

tensorflow-gpu v1.5.0 | cuda9.0 |  cuDNN  不明確 | 備註:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

tensorflow-gpu v1.4.0 | cuda8.0 |  cuDNN 6.0 | 備註:6.0正常使用, 7.0.5不能用,5.1未知 

tensorflow-gpu v1.3.0 | cuda8.0 |  cuDNN 6.0 | 備註:6.0正常使用, 7.0.5不能用,5.1未知 

tensorflow-gpu v1.2.0 | cuda8.0 |  cuDNN 5.1 | 備註:5.1正常使用, 6.0/ 7.0.5 未知

tensorflow-gpu v1.1.0 | cuda8.0 |  cuDNN 5.1 | 備註:5.1正常使用, 6.0/ 7.0.5 未知

 

P.S:這裡有可能會出現找不到cudnn64_7.dll的現象,解決方法如下:

cudnn64_7.dll  複製到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
cudnn.h 複製到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
cudnn.lib 複製到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64