深度學習圖像分類之下載安裝Pytorch

1. 簡述Pytorch

PyTorch是一個開源的Python機器學習庫,其前身是著名的機器學習庫Torch宁否。2017年1月副渴,由Facebook人工智能研究院(FAIR)基于Torch推出了PyTorch翼悴,它是一個面向Python語言的深度學習框架信卡,不僅能夠實現(xiàn)強大的GPU加速隔缀,同時還支持動態(tài)神經(jīng)網(wǎng)絡,這是很多主流深度學習框架比如Tensorflow等都不支持的傍菇。PyTorch既可以看作加入了GPU支持的numpy猾瘸,同時也可以看成一個擁有自動求導功能的強大的深度神經(jīng)網(wǎng)絡。除了Facebook外,它已經(jīng)被Twitter牵触、CMU和Salesforce等機構采用淮悼。作為經(jīng)典機器學習庫Torch的端口,PyTorch 為 Python 語言使用者提供了舒適的深度學習開發(fā)選擇揽思。

目前Pytorch成為深度學習領域使用最火熱的框架其原因主要包括以下三點:
(1) 簡潔袜腥; (2)速度; (3)易用绰更;

2. 安裝步驟:

2.1 安裝Python

由于Pytorch面向的是Python語言瞧挤,因此首先需要安裝Python。
由于之前已經(jīng)安裝過Python儡湾,因此這里不再重復安裝(我安裝的版本是3.7)。
Python的安裝可以參考:全網(wǎng)最詳細的Python安裝教程(Windows) - 知乎 (zhihu.com)

2.2 安裝CUDA

要特別注意的是Pytorch對CUDA的版本是有要求的执俩,因此我們安裝CUDA之前需要先確認到底裝哪個CUDA版本才行徐钠。
(1) 首先我們要確定本機是否有獨立顯卡。計算機->設備管理器->顯示適配器 中役首,查看是否有獨立顯卡尝丐,如下圖所示:

查看獨立顯卡

如上圖所示,可以看到衡奥,當前系統(tǒng)擁有NVIDIA顯卡爹袁,型號為GeForce GTX 960M。
(2) 其次我們要查看本機顯卡適合什么版本的CUDA矮固。NVIDIA控制面板->幫助->系統(tǒng)信息->組件

查看適合的CUDA

由上圖可知本機適合的CUDA版本為8.0失息。
(3) 開始安裝CUDA。
根據(jù)正常思路档址,我們進入CUDA安裝官網(wǎng)安裝:https://developer.nvidia.com/cuda-toolkit-archive盹兢,但是我發(fā)現(xiàn)該官網(wǎng)無論我用什么方法都無法進入,于是我參考該文章關于下載cuda和cudnn官方網(wǎng)站訪問不了的替代方案_sinat_29217765的博客-CSDN博客將網(wǎng)址https://developer.nvidia.com/cuda-toolkit-archive中的com改為cn后(https://developer.nvidia.cn/cuda-toolkit-archive)可順利訪問守伸。

CUDA官網(wǎng)

下面我們選擇 CUDA Toolkit 8.0進行下載绎秒。單擊后會進入版本配置界面,按下圖進行選擇即可:

下載信息選擇

下載完成后便自動進入安裝程序尼摹。

安裝

根據(jù)提示一步步安裝即可见芹,安裝完畢之后打開cmd命令終端測試一下cuda是否安裝成功
輸入命令測試:

 nvcc -V

正常情況下會出現(xiàn)下圖所示結果,說明安裝成功:

測試結果

2.3 安裝cuDNN蠢涝。

由于官網(wǎng)下載cuDnn需要加入Nvidia developers會員才能進行下載,因此參考文章官網(wǎng)安裝Nvidia CUDA Toolkit玄呛、cuDNN報錯502的備用鏡像下載網(wǎng)址_sinat_28371057的博客-CSDN博客_cudnn 鏡像進行下載。要注意的是CUDA8.0對應的cuDNN版本是5.1惠赫。
下載之后把鉴,解壓縮,將CUDNN壓縮包里面的bin、clude庭砍、lib文件直接復制到CUDA的安裝目錄下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)场晶,直接覆蓋安裝即可完成。

2.4 安裝Pytorch

安裝完CUDA和cuDNN后怠缸,安裝Pytorch相對比較簡單诗轻。(雖然我還是遇到了蠻多問題)

進入Pytorch的安裝選擇配置頁面,此時根據(jù)相關配置會自動給出安裝方法揭北,如下所示:Start Locally | PyTorch

Pytorch的安裝選擇配置頁面

但是由于我根據(jù)本機的配置安裝的版本為CUDA 8.0因此不在此頁面內(nèi)扳炬,我參考文章PyTorch 安裝(CUDA 是8.0以下舊版本,如何安裝 pytorch?)_u014546828的博客-CSDN博客進行安裝搔体,只需要在cmd終端中運行相應的命令即可:

pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

但是由于使用cmd終端進行pip下載的時候一直下載到一半卡住恨樟,于是參考文章Python:pip install torch官網(wǎng)速度太慢的解決辦法_誰偷了我的酒窩的博客-CSDN博客使用迅雷下載相應的安裝包。下載完成后疚俱,用命令行打開文件所在的路徑劝术,重新輸入安裝命令。

安裝torch

安裝torchvision的時候我看許多文章表示安裝完正確版本的torch后直接pip install就行呆奕,于是我照做了养晋,于是我出錯了,下面是我踩坑過程梁钾,請勿模仿

安裝torchvision

從上圖我們可以看到绳泉,系統(tǒng)發(fā)現(xiàn)我之前按照了torch0.4.1于是它把我的0.4.1版本刪了刪了刪了,換成了其他版本...我剛開始沒認真看于是接著驗證了。

2.5 驗證

驗證Pytorch是否安裝成功并且能夠成功調(diào)用cuDNN姆泻。

驗證Pytorch

使用命令行進行驗證后發(fā)現(xiàn)零酪,雖然安裝成功但是安裝的是cpu版本,因此導致驗證False麦射,問題應該是出在前面安裝完torch后采用自動安裝torchvision的時候它自動匹配選擇了版本蛾娶,因此我只能卸載重新安裝,卸載命令很簡單潜秋,就是把之前安裝時的install改成uninstall即可蛔琅。

卸載torch與torchvision

經(jīng)歷慘痛經(jīng)歷的我這次我根據(jù)文章pythorch版本和torchvision版本對應關系及torchvision安裝_村西那條彎彎的河流的博客-CSDN博客下載相應版本的torchvision再進行pip安裝。

版本對應

因此下載0.2.2版本的torchvision即可峻呛,但是在官網(wǎng)的鏈接里我并沒有找到這個版本的torchvision罗售,于是我在清華源找到了這個版本Links for torchvision (tsinghua.edu.cn)

清華源

找到鏈接后一樣的方法使用迅雷進行下載,速度超快钩述,幾秒鐘就完事了寨躁,然后打開下載路徑進行pip安裝即可。安裝完成后重新進行驗證牙勘。

驗證成功

歷經(jīng)磨難职恳!由圖可知所禀,版本正確,并且cuda正撤徘眨可用色徘,說明我們的Pytorch安裝成功告一段落,撒花操禀!

?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末褂策,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子颓屑,更是在濱河造成了極大的恐慌斤寂,老刑警劉巖,帶你破解...
    沈念sama閱讀 221,198評論 6 514
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件揪惦,死亡現(xiàn)場離奇詭異遍搞,居然都是意外死亡,警方通過查閱死者的電腦和手機器腋,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,334評論 3 398
  • 文/潘曉璐 我一進店門尾抑,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人蒂培,你說我怎么就攤上這事“裆唬” “怎么了护戳?”我有些...
    開封第一講書人閱讀 167,643評論 0 360
  • 文/不壞的土叔 我叫張陵,是天一觀的道長垂睬。 經(jīng)常有香客問我媳荒,道長,這世上最難降的妖魔是什么驹饺? 我笑而不...
    開封第一講書人閱讀 59,495評論 1 296
  • 正文 為了忘掉前任钳枕,我火速辦了婚禮,結果婚禮上赏壹,老公的妹妹穿的比我還像新娘鱼炒。我一直安慰自己,他們只是感情好蝌借,可當我...
    茶點故事閱讀 68,502評論 6 397
  • 文/花漫 我一把揭開白布昔瞧。 她就那樣靜靜地躺著,像睡著了一般菩佑。 火紅的嫁衣襯著肌膚如雪自晰。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,156評論 1 308
  • 那天稍坯,我揣著相機與錄音酬荞,去河邊找鬼。 笑死,一個胖子當著我的面吹牛混巧,可吹牛的內(nèi)容都是我干的枪向。 我是一名探鬼主播,決...
    沈念sama閱讀 40,743評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼牲剃,長吁一口氣:“原來是場噩夢啊……” “哼遣疯!你這毒婦竟也來了?” 一聲冷哼從身側響起凿傅,我...
    開封第一講書人閱讀 39,659評論 0 276
  • 序言:老撾萬榮一對情侶失蹤缠犀,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后聪舒,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體辨液,經(jīng)...
    沈念sama閱讀 46,200評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,282評論 3 340
  • 正文 我和宋清朗相戀三年箱残,在試婚紗的時候發(fā)現(xiàn)自己被綠了滔迈。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,424評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡被辑,死狀恐怖燎悍,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情盼理,我是刑警寧澤谈山,帶...
    沈念sama閱讀 36,107評論 5 349
  • 正文 年R本政府宣布,位于F島的核電站宏怔,受9級特大地震影響奏路,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜臊诊,卻給世界環(huán)境...
    茶點故事閱讀 41,789評論 3 333
  • 文/蒙蒙 一鸽粉、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧抓艳,春花似錦触机、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,264評論 0 23
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至庐椒,卻和暖如春椒舵,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背约谈。 一陣腳步聲響...
    開封第一講書人閱讀 33,390評論 1 271
  • 我被黑心中介騙來泰國打工笔宿, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留犁钟,地道東北人。 一個月前我還...
    沈念sama閱讀 48,798評論 3 376
  • 正文 我出身青樓泼橘,卻偏偏與公主長得像涝动,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子炬灭,可洞房花燭夜當晚...
    茶點故事閱讀 45,435評論 2 359

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