win10系統極簡安裝Tensorflow2.1—gpu版本流程

最近繼續研究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秒才行,但是第二次運行就是秒出,我估計是第一次會初始化很多東西導致的。所以耐心一點就行

發佈了51 篇原創文章 · 獲贊 25 · 訪問量 17萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章