pip3安裝Tensorflow-GPU

系統：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, 即安裝成功。