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