win10安裝TensorFlow(GPU)簡明教程

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章