TensorFlow-gpu在windows上的開發(fā)環(huán)境搭建

Step1:Nvidia顯卡驅(qū)動安裝

官網(wǎng)下載系統(tǒng)版本一致的顯卡驅(qū)動,針對安裝失敗的情況,需要退出殺毒軟件,重新安裝。

image.png

成功安裝后,桌面電腦->右擊->屬性->設備管理->顯示設備,可以看到NVIDIA TITAN Xp上面沒有感嘆號

image.png

Step2:安裝VS2017+Windows10 SDK**

安裝vs2017后通過Visual Studio Installer修改安裝

Step3:安裝CUDA10.0+cuDNN

Cuda的最新版本為cuda10.1, 一開始我安裝的是最新的cuda,但是安裝完TensorFlow-GPU后,出現(xiàn)了TensorFlow模塊導入失敗的錯誤(faile to load to load native tensorflow runtime)。卸載cuda10.1,重新安裝cuda10.0此問題可解決,此問題的原因可能是TensorFlow-gpu版本不支持最新版本的CUDA。

Nvidia官網(wǎng)選擇下載系統(tǒng)版本一致的CUDA版本

image.png

安裝完成,添加以下環(huán)境變量。

image.png

安裝完成后,在命令行輸入nvcc –v出現(xiàn)如下信息,則cuda安裝成功。

image.png

Step4:Python環(huán)境安裝

Anaconda是一個Python的科學計算發(fā)行版,包含了超過300個流行的用于科學、數(shù)學、工程和數(shù)據(jù)分析的Python Packages。由于Python有2和3兩個版本,因此Anaconda也在Python2和Python3的基礎上推出了兩個發(fā)行版,即Anaconda2Anaconda3。

針對開源資料要么使用Anaconda2就是Anaconda3,所以電腦上很有可能同時使用Anaconda2和Anaconda3兩個版本,因此需要同時安裝兩個版本。一般的使用場景是以其中一個版本為主,另外一個版本為輔,因此只需要達到能方便的切換到備用版本,且備用版本的pip等命令能正常使用即可。

操作步驟


  1. 首先安裝主版本,選擇Anaconda2,按照正常步驟安裝即可,這里假定安裝目錄為D:\Anaconda2。
image.png
  1. 在這一步打上那兩個勾。
image.png

第一個選項是將安裝目錄加入到系統(tǒng)的PATH環(huán)境變量中,以后在CMD中便可以直接用python命令啟動python。 第二個選項是讓其他IDE能夠檢測到Anaconda2并將Anaconda2作為默認的Python 2.7。

  1. 安裝好Anaconda2之后,再安裝Anaconda3,這里Anaconda3的安裝目錄必須選在D:\Anaconda2\envs子目錄下,如圖所示,最后的”py3”可以自己另取。
image.png
  1. 這里取消掉那兩個勾(按照字面意思理解,打上第二個勾應該是沒有影響的。至于第一個選項,由于之前已經(jīng)把Anaconda2添加到系統(tǒng)PATH路徑下了,因此再把Anaconda3添加進去,由于順序在Anaconda2的后面,幾乎是沒有作用的,所以這里沒有勾上)。
image.png
  1. 安裝完成之后,在CMD里面直接輸入python會啟動Python2,而使用activate py3(py3即之前Python3安裝目錄文件夾的名字)命令之后,再使用python即可切換至Python3,如下圖所示。使用activate py3命令之后,在命令行前面會出現(xiàn)一個[py3]標記,此時使用任何的python命令都是在Python3下進行的。使用deactivate命令可取消激活Python3
image.png

Step5:Tensorflow安裝

  1. 這里安裝基于python3.7的環(huán)境
image.png
  1. 安裝pip3
image.png
  1. 安裝虛擬環(huán)境,Python虛擬環(huán)境用于將軟件包安裝與系統(tǒng)隔離開來
image.png
  1. 安裝支持GPU的TensorFlow版本
pip3 install –upgrade tensorflow-gpu
image.png

安裝完成后,輸入以下兩行驗證TensorFlow-gpu是否能正常運行

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
image.png

紅線標識的第一行是一個警告,可以忽略;

紅線標識的第二行是因為CUDA未添加環(huán)境變量導致,按cuda部分安裝說明添加環(huán)境變量,可以解決該問題;

紅線標識的第三行提示顯卡設備映射問題,一開始我一直懷疑驅(qū)動版本問題,重復安裝了幾遍驅(qū)動都不行,后來通過重新回到第一步找原因,發(fā)現(xiàn)顯卡設備未正常工作,開始懷疑硬件原因,重新動了動兩塊顯卡及顯卡間的搭橋。重新開機,桌面電腦右擊屬性->設備管理->顯示設備兩塊顯卡并沒有感嘆號。

而且運行CUDA安裝目錄下的deviceQuery.exe,能正常顯示兩塊顯卡設備信息。

image.png

再次使用以下這兩行,也能顯示正常運行:

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
image.png

至此,TensorFlow-gpu在windows上的開發(fā)環(huán)境搭建徹底完成

需注意事項:

  1. 每安裝完一項,驗證一項是否安裝成功,確保后徐的環(huán)境搭建順利;

  2. 注意TensorFlow和Cuda版本的兼容性問題;

  3. 注意顯卡驅(qū)動和顯卡的兼容性問題(官網(wǎng)下載相應顯卡的最新驅(qū)動,安裝的時候采用執(zhí)行清潔安裝,一般不會出現(xiàn)任何問題)。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容