安裝gpu版本首先是英偉達的顯卡。
需要用到的軟件有:
1.python 3.6
2.Tensorflow 1.11
3.CUDA:cuda_8.0_win10.exe
4.cuDNN: cudnn-8.0-windows10-x64-v6.0.zip
讓我們開始吧~
1.首先,下載cuda8.0版本以及cudnn6.0版本(cuda和cudnn的版本一定要對應)。
cuda8.0下載鏈接:https://developer.nvidia.com/cuda-80-download-archive
cudnn6.0下載鏈接:https://developer.nvidia.com/rdp/cudnn-archive
2.讓我們安裝cuda8.0
點擊運行下載好的cuda8.0.exe。會出現下圖,注意!!!!這個文件夾是安裝過程中的臨時文件存儲的目錄,並不是最終的安裝目錄(默認安裝的主目錄是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)。后邊就很簡單了,一步步安裝好。
讓我們驗證以下是不是安裝成功了~
打開cmd,輸入nvcc -V
如圖所示,我們的cuda已經安裝好了,並且顯示了版本號8.0.6
3.安裝cuDNN6.0
把cudnn6.0.zip解壓如下圖
然后找到cuda8.0的主目錄(也就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0),將解壓出來的三個子文件copy到該目錄下,成功的話能看到cudnn64_6.dll
4.安裝tensorflow(以anaconda為例)
首先創建tensorflow gpu版本的環境。在anaconda prompt中輸入:
conda create -n tensorflow-gpu python=3.6
然后激活:
activate tensorflow-gpu
最后就是安裝:
conda install tensorflow-gpu
等待結束就可以啦。
5.在pycharm中使用
file->setting 解釋器一定要選anaconda\envs\tensorflow-gpu\python.exe要不在pycharm中會報錯
6.測試一下吧
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
如圖所示,測試程序成功!!!!
tensorflow你就可以用啦。