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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM