【深度之眼PyTorch框架班三期】pytorch安裝 GPU版本(win10,GTX1060墅冷,CUDA9.2)

(之前為了安裝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安裝成功

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市稿壁,隨后出現(xiàn)的幾起案子幽钢,更是在濱河造成了極大的恐慌,老刑警劉巖傅是,帶你破解...
    沈念sama閱讀 222,865評(píng)論 6 518
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件搅吁,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡落午,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,296評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門(mén)肚豺,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)溃斋,“玉大人,你說(shuō)我怎么就攤上這事吸申」=伲” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 169,631評(píng)論 0 364
  • 文/不壞的土叔 我叫張陵截碴,是天一觀(guān)的道長(zhǎng)梳侨。 經(jīng)常有香客問(wèn)我,道長(zhǎng)日丹,這世上最難降的妖魔是什么走哺? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 60,199評(píng)論 1 300
  • 正文 為了忘掉前任,我火速辦了婚禮哲虾,結(jié)果婚禮上丙躏,老公的妹妹穿的比我還像新娘择示。我一直安慰自己,他們只是感情好晒旅,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,196評(píng)論 6 398
  • 文/花漫 我一把揭開(kāi)白布栅盲。 她就那樣靜靜地躺著,像睡著了一般废恋。 火紅的嫁衣襯著肌膚如雪谈秫。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 52,793評(píng)論 1 314
  • 那天鱼鼓,我揣著相機(jī)與錄音拟烫,去河邊找鬼。 笑死蚓哩,一個(gè)胖子當(dāng)著我的面吹牛构灸,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播岸梨,決...
    沈念sama閱讀 41,221評(píng)論 3 423
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼喜颁,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了曹阔?” 一聲冷哼從身側(cè)響起半开,我...
    開(kāi)封第一講書(shū)人閱讀 40,174評(píng)論 0 277
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎赃份,沒(méi)想到半個(gè)月后寂拆,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,699評(píng)論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡抓韩,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,770評(píng)論 3 343
  • 正文 我和宋清朗相戀三年纠永,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片谒拴。...
    茶點(diǎn)故事閱讀 40,918評(píng)論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡尝江,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出英上,到底是詐尸還是另有隱情炭序,我是刑警寧澤,帶...
    沈念sama閱讀 36,573評(píng)論 5 351
  • 正文 年R本政府宣布苍日,位于F島的核電站惭聂,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏相恃。R本人自食惡果不足惜辜纲,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,255評(píng)論 3 336
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧侨歉,春花似錦屋摇、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 32,749評(píng)論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至牵舵,卻和暖如春柒啤,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背畸颅。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 33,862評(píng)論 1 274
  • 我被黑心中介騙來(lái)泰國(guó)打工担巩, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人没炒。 一個(gè)月前我還...
    沈念sama閱讀 49,364評(píng)論 3 379
  • 正文 我出身青樓涛癌,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親送火。 傳聞我的和親對(duì)象是個(gè)殘疾皇子拳话,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,926評(píng)論 2 361

推薦閱讀更多精彩內(nèi)容