(之前為了安裝tensorflow-gpu纯路,看了網(wǎng)上的很多教程,都寫(xiě)的五花八門(mén)的寞忿。雖然最后是裝成功了驰唬,但是真的費(fèi)了很大的勁,花了很多時(shí)間腔彰。個(gè)人的經(jīng)驗(yàn)是叫编,GPU,CUDA萍桌,cudnn還有各種包的版本之間的對(duì)應(yīng)很重要宵溅,所以為了安裝成功,最好是找一個(gè)已經(jīng)安裝成功上炎,并且和你的機(jī)器恃逻、操作系統(tǒng)、硬件藕施、軟件等都非常相似的人寫(xiě)的教程寇损,照著安裝,這樣成功的幾率會(huì)比較大(如果有更多的需求可以考慮容器)裳食。這次看了深度之眼余老師的視頻矛市,發(fā)現(xiàn)他把CUDA,cudnn诲祸,pytorch的安裝講的很簡(jiǎn)單浊吏,真的是對(duì)我這種小白很友好而昨,所以記錄下)
首先說(shuō)明,這里主要是要安裝GPU版本的pytorch找田,還附帶了conda虛擬環(huán)境安裝和pycharm安裝歌憨。我用的操作系統(tǒng)是win10,安裝的CUDA版本是9.2墩衙,顯卡是GTX1060务嫡。
下面就開(kāi)始吧:(在這之前最好確定已經(jīng)安裝好了顯卡的驅(qū)動(dòng))總體的步驟為:Anaconda安裝->pycharm安裝->CUDA安裝->cudnn安裝->pytorch安裝(含conda虛擬環(huán)境安裝来氧,pycharm安裝)
一傻谁、Anaconda安裝
Anaconda下載地址:https://www.anaconda.com/distribution/#download-section
選windows下的python3.7的64-Bit下載谆甜。
下載好以后辽狈,雙擊exe文件直接安裝。安裝過(guò)程中一直用它默認(rèn)的選項(xiàng)就好了鸠蚪,需要注意的是孙咪,在下面這一步宁改,這里要注意選擇把Anaconda加入到系統(tǒng)環(huán)境變量樊破。
然后可以查看anaconda是否安裝成功厅篓,打開(kāi)cmd,輸入conda捶码,回車(chē)。
如果出現(xiàn)類(lèi)似下面的信息或链,就是安好了惫恼。
二、pycharm安裝
去pycharm官網(wǎng)下載→https://www.jetbrains.com/pycharm/download/#section=windows
下載windows的Professional版本澳盐。(Professional版要破解祈纯,如果不想破解,下載免費(fèi)的Community版就好了叼耙,兩者差別很小腕窥。)
然后雙擊exe文件安裝,安裝過(guò)程中一直用它默認(rèn)的選項(xiàng)就好了筛婉,需要注意的是這一步簇爆,這里要注意選擇把pycharm加入到環(huán)境變量。
因?yàn)橄碌氖荘rofessional版爽撒,要破解入蛆,至于怎么破解,也挺簡(jiǎn)單的硕勿,自行百度谷歌下吧哨毁,這里就不寫(xiě)了。
三源武、CUDA安裝
因?yàn)檫@里是為了安裝pytorch扼褪,先去pytorch的官網(wǎng)查看pytorch支持的CUDA版本→https://pytorch.org/get-started/locally/
從下圖可以看到想幻,pytorch支持CUDA9.2和10.1,所以這里選擇下載CUDA9.2话浇。
CUDA9.2的下載地址→https://developer.nvidia.com/cuda-92-download-archive
點(diǎn)windows脏毯,x86_64,10凳枝,exe(local)抄沮,選base installer的download,下載岖瑰。
雙擊exe安裝
下面檢查CUDA是否安裝成功
使用cmd進(jìn)入這個(gè)路徑(實(shí)際上就是CUDA安裝在你電腦里面的路徑叛买,因人而異)
執(zhí)行命令: nvcc -V
出現(xiàn)下面類(lèi)似的結(jié)果,就是安裝成功了
四蹋订、cudnn安裝
cudnn下載網(wǎng)址→https://developer.nvidia.com/rdp/cudnn-download率挣。cudnn需要登錄賬號(hào)才可以下載,可以直接注冊(cè)一個(gè)賬號(hào)露戒。
注冊(cè)好賬號(hào)然后登陸椒功,然后就可以下載了。下面開(kāi)始下載和安裝智什。
勾選下面圖中這個(gè)
點(diǎn)擊如下圖所示的地方進(jìn)行下載动漾。
把這個(gè)下載好的cudnn壓縮包解壓,將解壓后里面的三個(gè)文件夾
粘貼到之前的CUDA的文件夾下面荠锭,如下圖所示旱眯。
下面檢查cudnn是否安裝成功
用命令行cmd進(jìn)入下圖路徑(下圖這個(gè)路徑在之前說(shuō)的那個(gè)CUDA所在的路徑的下面,仔細(xì)找找能找到的)
輸入命令:bandwidthTest.exe
如果出現(xiàn)pass就表示安裝成功
也可以輸入命令:deviceQuery.exe
出現(xiàn)了下圖的CUDA版本號(hào)证九,GPU型號(hào)删豺,還有PASS等信息,就說(shuō)明CUDA和cudnn安裝成功了愧怜。
五呀页、pytorch安裝(含conda虛擬環(huán)境安裝,pycharm安裝)
先下載pytorch
pytorch下載地址→https://pytorch.org/get-started/locally/
選下圖這些
復(fù)制這個(gè)網(wǎng)址拥坛,進(jìn)入
下載這兩個(gè)文件蓬蝶,文件名中的cp37代表對(duì)應(yīng)python3.7
下面創(chuàng)建虛擬環(huán)境
打開(kāi)pycharm,點(diǎn)最下面的Terminal
輸入這個(gè)命令用以創(chuàng)建虛擬環(huán)境:conda create -n pytorch_gpu python=3.7
上面的命令中的pytorch_gpu是你給你的虛擬環(huán)境取的名字猜惋,不一定叫這個(gè)名字疾党,可以自己取
在創(chuàng)建的過(guò)程中會(huì)有一條下圖這樣的提示,這是虛擬環(huán)境保存的地址惨奕,要記錄雪位,等下會(huì)用到。
創(chuàng)建好之后梨撞,輸入命令conda activate pytorch_gpu進(jìn)入虛擬環(huán)境
出現(xiàn)這個(gè)就說(shuō)明進(jìn)入虛擬環(huán)境了
下面在剛才建立的虛擬環(huán)境中安裝pytorch
還是在之前那個(gè)地方雹洗,輸入命令 cd XXXXXXXXX
命令中的XXXXXXX表示你自己pytorch的下載后的保存路徑
輸入 pip install torch香罐,然后按下Tab鍵
命令自動(dòng)補(bǔ)全為下圖,然后回車(chē)安裝时肿。
成功后會(huì)出現(xiàn)如下提示
輸入 pip install torchvision庇茫,然后按下Tab鍵
命令自動(dòng)補(bǔ)全為下圖,然后回車(chē)安裝螃成。
然后檢查pytorch是否安裝正確
先新建一個(gè)pycharm項(xiàng)目旦签,如何新建可以自行百度
然后把項(xiàng)目和剛剛安好pytorch的虛擬環(huán)境關(guān)聯(lián)上,如下圖寸宏,進(jìn)入pycharm的設(shè)置
如圖點(diǎn)擊
然后如下圖點(diǎn)擊
進(jìn)入之前記錄的你虛擬環(huán)境保存的地址宁炫,點(diǎn)python.exe,此時(shí)紅框中就會(huì)出現(xiàn)類(lèi)似的地址氮凝。然后點(diǎn)OK
等待一會(huì)羔巢,就關(guān)聯(lián)成功了。
然后檢查pytorch是否安裝成功
在這個(gè)pycharm項(xiàng)目中罩阵,新建一個(gè)python file竿秆,輸入如下命令:
import torch
print("pytorch version:{}".format(torch.version)) # 查看pytorch 版本
print(torch.cuda.is_available()) # 查看是否支持GPU
如果出現(xiàn)提示:
說(shuō)明pytorch安裝成功