1. 安裝準備
(1)CUDA 9.0
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64
建議選擇 exe(local) 文件下載
注:在Anoconda中安裝的tensorflow目前只支持到CUDA 9.0,更高版本會報錯。如有需要可通過Bazel重新編譯爲.whl文件安裝
(2)cuDNN 7.3.1
https://developer.nvidia.com/cudnn (需要註冊一下)
2. 安裝CUDA和cuDNN
(1)安裝CUDA
按照提示的步驟來,保持默認安裝路徑即可.
唯一要注意的地方就是,cuda中的某些驅動版本可能較低,這時候就不要勾選該安裝項
(2)安裝cuDNN
將解壓後得到三個文件夾複製到CUDA的安裝目錄下對應的文件下即可
默認路徑爲:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
- lib
- bin
- include
(3)驗證
win+R
->cmd
鍵入nvcc -V
,可以看到如下輸出,則說明安裝成功
3. 安裝TensorFlow
同樣通過win+R
->cmd
打開終端鍵入
conda create -n tensorflow-gpu python=3.6
接着輸入
activate tensorflow-gpu
激活剛剛安裝完成的環境,不需要時通過deactivate tensorflow-gpu
關閉該環境
4. 測試
在激活環境的基礎上,通過python
進入python終端
複製代碼到終端中
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)
可以看到如下輸出
b'Hello, TensorFlow!'
至此,TensorFlow安裝完成。可以愉快的玩耍了~~~o(∩_∩)o
在VSCode中搭建TensorFlow可以參見下篇博文:
https://blog.csdn.net/Briliantly/article/details/82847804