|
一、明確前提 首先我們安裝的PyTorch是cuda版本的，所以首先要知道你的cuda版本（我的cuda是 10.2） 二、打開pytorch官網(wǎng)，查看安裝命令 https://pytorch.org/get-started/locally/
這里我沒有安裝最新版本怕不穩(wěn)定
選擇一個適合你的版本（先別急輸入命令哦！） a）官網(wǎng)安裝 conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch b）鏡像安裝（去掉 -c ） conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 pytorch 確定你已經(jīng)配置過pytorch的國內(nèi)鏡像，如果使用國內(nèi)鏡像安裝不成功就多試兩次，總是不行可以選擇官網(wǎng)安裝（親測！有時官網(wǎng)安裝也不慢哦！）
三、創(chuàng)建新的anaconda環(huán)境 conda create -n python36torch15 python=3.6
環(huán)境創(chuàng)建好之后，輸入之前找到的安裝pytorch的命令
四、驗(yàn)證安裝 那你的cuda版本的pytorch安裝好了沒有呢？見證奇跡的時刻到了。 在你創(chuàng)建的anaconda環(huán)境中（我的是python36torch15）輸入 python 進(jìn)入python代碼編寫環(huán)境 輸入代碼 #判斷是否安裝了cuda
import torch
print(torch.cuda.is_available()) #返回True則說明已經(jīng)安裝了cuda
—————— 到這里pytorch就裝好了開始你的pytorch之旅吧 ——————
****** 下面這個是贈送的福利哦！相信你后面會用到的 ****** #判斷是否安裝了cuDNN from torch.backends import cudnn print(cudnn.is_available()) #返回True則說明已經(jīng)安裝了cuDNN
最后~祝科研順利啊！！！ |
|
|
