linux環(huán)境GPU版pytorch安裝教程

一葵姥、 安裝python環(huán)境涌庭,建議使用anaconda

Anaconda

Anaconda是一個(gè)開源的Python發(fā)行版本滞项,包含大量科學(xué)計(jì)算相關(guān)包伤哺。

Anaconda提供包管理和環(huán)境管理功能猛遍,可以方便解決多版本python并存顶捷,切換臊诊,軟件包極其依賴安裝問(wèn)題口芍。

官網(wǎng) : https://www.anaconda.com/

image.png

進(jìn)入download頁(yè)面選擇對(duì)應(yīng)python版本下載杖剪。

Anaconda服務(wù)器在海外蠢笋,速度較慢拨齐,可以到清華鏡像網(wǎng)搜索Anaconda下載。

https://mirrors.tuna.tsinghua.edu.cn/

Anaconda自帶很多科學(xué)計(jì)算相關(guān)包了昨寞,如果需要的包沒有瞻惋,可以用pip或conda安裝。

如要安裝numpy援岩,終端輸入

pip install numpy 或者 conda install numpy 熟史。

如果覺得下載慢,可以用清華鏡像窄俏。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple your-package

二蹂匹、pytorch安裝

PyTorch

由Torch7團(tuán)隊(duì)開發(fā),以python為開發(fā)語(yǔ)言的深度學(xué)習(xí)框架凹蜈,能實(shí)現(xiàn)GPU加速和動(dòng)態(tài)神經(jīng)網(wǎng)絡(luò)(很多主流框架如Tensorflow不支持)

PyTorch可看做GPU加速支持的numpy限寞,也可看做擁有自動(dòng)求導(dǎo)功能的深度神經(jīng)網(wǎng)絡(luò)忍啸。

CPU版本的pytorch的安裝

進(jìn)入pytorch官網(wǎng) : https://pytorch.org/

image.png

選擇要安裝的版本,如果電腦沒有可進(jìn)行GPU加速的顯卡(比如我的macbook)履植,cuda就選擇none计雌。
在終端運(yùn)行下方生成的語(yǔ)句。

conda install pytorch torchvision -c pytorch

如果電腦有顯卡可GPU加速玫霎,需要先安裝cuda凿滤,才能安裝GPU版本的pytorch

GPU版本的pytorch的安裝

1. 安裝 Nvidia Cuda

首先確認(rèn)電腦顯卡安裝好驅(qū)動(dòng)且支持cuda。

linux顯卡驅(qū)動(dòng)安裝庶近。
進(jìn)入系統(tǒng)設(shè)置


image.png

image.png

選擇相應(yīng)顯卡驅(qū)動(dòng)并安裝翁脆。

安裝 Nvidia Cuda :

https://developer.nvidia.com/cuda-downloads

image.png

根據(jù)下方給出的命令在終端運(yùn)行。

image.png

最好opengl選擇n鼻种,因?yàn)榭赡苡邪惭b后無(wú)法啟動(dòng)圖形化桌面的問(wèn)題反番。

配置環(huán)境變量
sudo vim ~/.bash_profile

在文本末尾加上

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
2. 安裝 CuDNN

進(jìn)入 : https://developer.nvidia.com/cudnn
需要注冊(cè),注冊(cè)完搜索cudnn進(jìn)入如下界面:

image.png

選擇你要的版本叉钥,下載下來(lái)的是deb包
運(yùn)行sudo dpkg -i <package.deb>安裝

3. 安裝GPU版本pytorch

和之前安裝CPU版pytorch差不多罢缸,最后一行選擇cuda 8.0,執(zhí)行相關(guān)命令就行投队。

4. 測(cè)試

終端進(jìn)入python界面

>>> import torch
>>> torch.cuda.is_available()
True

結(jié)果為True枫疆,則GPU版pytorch安裝成功。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末敷鸦,一起剝皮案震驚了整個(gè)濱河市养铸,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌轧膘,老刑警劉巖钞螟,帶你破解...
    沈念sama閱讀 219,539評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異谎碍,居然都是意外死亡鳞滨,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,594評(píng)論 3 396
  • 文/潘曉璐 我一進(jìn)店門蟆淀,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)拯啦,“玉大人,你說(shuō)我怎么就攤上這事熔任“矗” “怎么了?”我有些...
    開封第一講書人閱讀 165,871評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵疑苔,是天一觀的道長(zhǎng)甫匹。 經(jīng)常有香客問(wèn)我,道長(zhǎng),這世上最難降的妖魔是什么兵迅? 我笑而不...
    開封第一講書人閱讀 58,963評(píng)論 1 295
  • 正文 為了忘掉前任抢韭,我火速辦了婚禮,結(jié)果婚禮上恍箭,老公的妹妹穿的比我還像新娘刻恭。我一直安慰自己,他們只是感情好扯夭,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,984評(píng)論 6 393
  • 文/花漫 我一把揭開白布鳍贾。 她就那樣靜靜地躺著,像睡著了一般交洗。 火紅的嫁衣襯著肌膚如雪骑科。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,763評(píng)論 1 307
  • 那天藕筋,我揣著相機(jī)與錄音纵散,去河邊找鬼梳码。 笑死隐圾,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的掰茶。 我是一名探鬼主播暇藏,決...
    沈念sama閱讀 40,468評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼濒蒋!你這毒婦竟也來(lái)了盐碱?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,357評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤沪伙,失蹤者是張志新(化名)和其女友劉穎瓮顽,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體围橡,經(jīng)...
    沈念sama閱讀 45,850評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡暖混,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,002評(píng)論 3 338
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了翁授。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片拣播。...
    茶點(diǎn)故事閱讀 40,144評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖收擦,靈堂內(nèi)的尸體忽然破棺而出贮配,到底是詐尸還是另有隱情,我是刑警寧澤塞赂,帶...
    沈念sama閱讀 35,823評(píng)論 5 346
  • 正文 年R本政府宣布泪勒,位于F島的核電站,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏酣藻。R本人自食惡果不足惜曹洽,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,483評(píng)論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望辽剧。 院中可真熱鬧送淆,春花似錦、人聲如沸怕轿。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,026評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)撞羽。三九已至阐斜,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間诀紊,已是汗流浹背谒出。 一陣腳步聲響...
    開封第一講書人閱讀 33,150評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留邻奠,地道東北人笤喳。 一個(gè)月前我還...
    沈念sama閱讀 48,415評(píng)論 3 373
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像碌宴,于是被迫代替她去往敵國(guó)和親杀狡。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,092評(píng)論 2 355

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