系統:win7
Python 3.6
安裝方法:直接使用pip3 安裝
官網上有詳細的要求及安裝方法:https://tensorflow.google.cn/install/gpu
首先,我們可以看到安裝GPU版需要電腦軟硬件滿足一定的條件,即需要首先安裝CUDA和cudnn, 這個直接點擊藍色的字體便能進入到下載界面,按順序下載安裝即可。
1) 下載安裝CUDA
https://developer.nvidia.com/cuda,ANZHAO 選擇自己需要的版本下載,下載完成后直接安裝,安裝步驟很簡單,記一下安裝路徑,后面會用到。
2) 下載安裝cudnn
https://developer.nvidia.com/cudnn
這個下載前需要先注冊一個賬戶,但是也沒有特殊要求,就簡單注冊一下就好了。
下載完成后解壓文件(看一個帖子里說要把壓縮包的擴展名改為.tgz格式在解壓,我直接照做了),解壓后的文件夾名為cuda,文件內容如下圖
將其中的文件拷貝到先前cuda 安裝路徑下的對應文件夾中
3) 在環境變量中添加路徑
將cuda安裝目錄下的bin和lib\x64兩個文件夾的路徑添加到環境變量PATH里面
4) pip3 安裝tensorflow-gpu
打開cmd窗口,輸入pip3 install tensorflow-gpu
安裝成功后會顯示 successfully
5) 安裝完成,測試。
編寫簡單代碼,測試tensorflow是否可以正常使用。
import tensorflow as tf
hello=tf.constant('hello')
sess=tf.Session()
print (sess.run(hello))
輸出”hello”且未報錯即安裝成功。
此外,還可使用一下代碼驗證:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
在結果中可以看到GPU, 即安裝成功。