最近繼續研究tensorflow,本打算後續上gpu版本的,但是運行demo過程中,實在太慢受不了了,就研究裝了一下gpu版本,現在把具體方法記錄如下:
一、安裝條件
可以參考tensorflow官網需求https://tensorflow.google.cn/install/gpu
1、nvidia gpu驅動必須在 418.x或者更高,可以用nvidia-smi命令查看
2、cuda toolkit 10.1版本支持tensorflow 2.1以上版本,我裝tensorflow2.1所以要cuda 10.1
3、cuDNN SDK 大於或等於7.6
以上條件需要滿足,但是有幾個極簡的方法可以完成安裝我實測也是可以的。
二、安裝步驟
1、安裝anaconda,不細說了
2、最好新建個環境,也不多說可以百度或看我之前文章
3、然後直接用下面命令即可:
conda install tensorflow-gpu==2.1.0
你會看到會自動下載相關依賴包,包括cuda,cudnn
注意:因爲我之前沒看到這個方法所以提前手動安裝了cuda和cudnn,但我看上面命令也會下載cuda和cudnn,應該可以直接用,我沒有親自實踐,應該沒問題
三、驗證是否安裝成功
用如下命令驗證是否gpu版本啓用
import tensorflow as tf
tf.test.is_gpu_available()
這裏也要注意一點:第一次運行這個命令是要等很久,看我下面圖上我這邊等了5分12秒才行,但是第二次運行就是秒出,我估計是第一次會初始化很多東西導致的。所以耐心一點就行