Pytorch安裝
一戏自、安裝anaconda
網(wǎng)址:https://www.anaconda.com/products/individual
網(wǎng)頁(yè)翻到最下面抑堡,選擇下載Windows 32位/64位版本骇径,傻瓜式安裝舶沛,記得勾選【Add to Path】 可以將anaconda的環(huán)境安裝到系統(tǒng)中去
二、Anaconda創(chuàng)建虛擬環(huán)境
1 、從開始菜單中打開Anaconda Navigator窍荧,單機(jī)進(jìn)入Environments選項(xiàng)卡,如下圖所示
2恨憎、在Environments選項(xiàng)卡中蕊退,單擊左下方的Create按鈕,創(chuàng)建虛擬環(huán)境憔恳,如下圖所示
此時(shí)在Environments選項(xiàng)卡中瓤荔,已經(jīng)出現(xiàn)剛剛創(chuàng)建的虛擬環(huán)境 【注】才創(chuàng)建的虛擬環(huán)境只包含最基本的庫(kù)
三、安裝Pycharm,創(chuàng)建工程
網(wǎng)址:https://www.jetbrains.com/pycharm/
1钥组、單擊download進(jìn)入下載頁(yè)面
此時(shí)输硝,有兩種版本選擇,嫌麻煩就用社區(qū)版者铜,但是功能較少腔丧,專業(yè)版需要自己找破解,或者用校園郵箱申請(qǐng)一年免費(fèi)作烟,一年后再次審查通過(guò)還可以繼續(xù)用
2愉粤、打開PyCharm,單擊New project
3拿撩、PyCharm創(chuàng)建Conda工程
4衣厘、在Anaconda安裝文件夾下找到envs文件夾,里面存儲(chǔ)的都是你創(chuàng)建的虛擬環(huán)境压恒,然后選擇剛剛創(chuàng)建的虛擬環(huán)境文件下的python.exe影暴,點(diǎn)擊OK
5、創(chuàng)建完成后自動(dòng)進(jìn)入如下界面:
四探赫、CUDA安裝
1型宙、查看CUDA支持版本
打開NVIDIA控制面板,按照?qǐng)D示順序查看CUDA所支持的最高版本
2伦吠、下載CUDA
打開網(wǎng)址:https://developer.nvidia.com/cuda-toolkit-archive妆兑,選擇適合自己電腦的cuda版本,比剛才查看的所支持的CUDA 版本低都行
此時(shí)要根據(jù)自己的系統(tǒng)進(jìn)行選擇
3毛仪、下載cuDNN
打開網(wǎng)址:https://developer.nvidia.com/cudnn搁嗓,點(diǎn)擊Download cuDNN
注意,cuDNN下載需要NVIDIA賬戶箱靴,沒(méi)有則先注冊(cè)腺逛,此步需要耐心,賊卡
4衡怀、登陸之后棍矛,點(diǎn)擊Download cuDNN進(jìn)入的應(yīng)該是如下界面安疗,先把問(wèn)卷填了,才能進(jìn)入下載界面
此時(shí)終于進(jìn)入下載界面了茄靠,勾選I agree…茂契,然后按照下圖操作
5蝶桶、選擇適合自己的操作系統(tǒng)版本下載
6慨绳、安裝CUDA
雙擊安裝包開始安裝,此時(shí)讓你選擇的路徑并不是安裝路徑真竖,而是暫時(shí)解壓出來(lái)的文件所需要存放的地方脐雪,安裝完成后會(huì)刪除,只要保證選擇空間足夠大的地方即可恢共,安裝路徑是默認(rèn)C盤里战秋,不能修改,此處可以不更改讨韭,點(diǎn)擊OK即可
驗(yàn)證CUDA是否安裝成功脂信,打開cmd,輸入 nvcc –V透硝,如下返回則說(shuō)明安裝成功
解壓壓縮包狰闪,壓縮包內(nèi)文件如下:
7、安裝cuDNN
打開cuda安裝文件夾濒生,默認(rèn)的強(qiáng)制安裝路徑為:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\版本號(hào)
打開安裝路徑如下:
將解壓的cuDNN壓縮包內(nèi)的三個(gè)文件夾復(fù)制到CUDA安裝目錄下:
驗(yàn)證cuDNN是否安裝完成埋泵,打開cmd,輸入
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite
然后執(zhí)行命令:
bandwidthTest.exe
然后再執(zhí)行命令:
deviceQuery.exe
8罪治、下載PyTorch安裝文件
打開網(wǎng)址:https://download.pytorch.org/whl/torch_stable.html丽声,界面如下:
每個(gè)文件命名都是有規(guī)律的:
使用快捷鍵Ctrl+F,搜索我們需要安裝的版本文件下載
五觉义、安裝pytorch
1雁社、打開PyCharm,進(jìn)入剛剛創(chuàng)建的工程里晒骇,打開Terminal窗口:
2霉撵、在PyCharm的Terminal窗口中鍵入cd 命令,切換目錄到PyTorch安裝包所在的地址厉碟,然后分別執(zhí)行pip install命令安裝兩個(gè)文件喊巍,先安裝torch,再安裝torchvision箍鼓,期間會(huì)一起下載所需要的依賴包
3崭参、現(xiàn)在驗(yàn)證是否安裝成功,在main.py里寫入以下代碼并運(yùn)行:
以上方式針對(duì)的是windows10操作系統(tǒng)款咖,大家可以根據(jù)大家各自的操作系統(tǒng)選擇何暮,安裝cuda或者不安裝cuda