win10 Tensorflow-gpu搭建環(huán)境

自己記錄下搭建環(huán)境的過程方便日后重新搭建

首先下載我們的python環(huán)境,這個了解的都知道anaconda這個東西吧不用多說,各種深度學(xué)習(xí)所需的包都在里面。

anaconda

下載頁面

一路安裝,注意一點就好它安裝的選項里有添加path環(huán)境變量給他打上勾。

安裝完后會發(fā)現(xiàn)path里會多


如果忘記添加就自己手動添加一下

安裝完成后下一步安裝tensorflow

But!

發(fā)現(xiàn)tensorflow官網(wǎng)上說明不支持python3.7.。。。


還能說啥

我們發(fā)現(xiàn)tf-gpu支持都是3.5或3.6

那還說啥當(dāng)然用最新的啊!!!果斷用3.6python

打開CMD輸入nb的命令

-------->? conda create --name py36 python=3.6

輸入后會自動下載安裝python3.6版本

在自己安裝的anaconda的目錄下x:xxx\Anaconda3\envs目錄下會生成我們上面命令生成文件py36,這個目錄下就是我們的python3.6環(huán)境

開始菜單打開anaconda prompt命令行


開始菜單

輸入命令:activate py36? ?就可以進(jìn)入python3.6的環(huán)境,不信的可以輸入python看看


anaconda prompt

到此裝好python環(huán)境開始下一個惡心的階段gpu環(huán)境


根據(jù)上面tensorflw官方提供說明我們需要cuda9.0 和 cudnn7

此處有一坑需要注意下,桌面鼠標(biāo)右鍵打開NVIDIA 控制面板-->幫助-->系統(tǒng)信息-->組件

這里可以看到我們能使用的cuda和cudnn版本,按照自己對應(yīng)的下載就可以了。

cuda下載??cudnn下載? ?cudnn需要注冊填寫資料自己弄弄就好



注意cuda和cudnn版本需要對應(yīng)? 如cuda10.0 對應(yīng)cudnn7.4.1

下完完成后安裝cuda 一路安裝沒啥特別注意的。

安裝完成后查看本機(jī)環(huán)境變量


我們可以看到cuda安裝的位置? 然后將我們下載的cudnn壓縮包解壓出來吧對應(yīng)的文件復(fù)制到cuda文件夾下

注意看第三的環(huán)境變量


是不是發(fā)現(xiàn)沒有啊。這個是需要自己手動設(shè)置的,一定要設(shè)置不然tf可是跑不起來的。


到此準(zhǔn)備環(huán)境都已經(jīng)差不多了,接下來重頭戲 T T T Tensorflow-gpu啊

此步驟是最雞兒簡單的

打開我們上面安裝的anaconda?prompt命令行運行我們py36的環(huán)境

在py36環(huán)境下輸入pip install tensorflow-gpu==x.x.x

輸入自己安裝的cuda所支持的tf版本可參考上面給出的版本適用圖


如果沒報錯那么說明安裝沒問題那么我們跑個官方代碼試試

import tensorflow astf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()print(sess.run(hello))

這里會打印Hello, TensorFlow!這就恭喜你可以開始編寫TF代碼了

到此環(huán)境搭建已經(jīng)完成。

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

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

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