剛剛成功裝上了PyTorch,先說一下我的環境,windows10 + Pycharm + python 3.7 + Anaconda3
這裏默認大家已經安裝了Anaconda3+Pycharm+python 3.7,如果沒有的話可以參考https://blog.csdn.net/CSS360/article/details/88215036
首先第一步,查看自己電腦的python版本,直接控制檯進去輸入python、
第二步,登錄PyTorch官網,https://pytorch.org/ 在下面選擇自己想要的版本,這是我的選擇
第一行大多都是Stable,第二行選擇自己的操作系統,第三行選擇安裝方式,因爲大家都安裝了Anaconda3,所以推薦Conda,第四行選擇自己的python版本,第五行是你的CUDA版本,關於查看自己電腦是否支持CUDA,大家參考https://www.jianshu.com/p/d3b9419a0f89 我的版本是10.2所以我這裏選擇10.0
緊接着選擇好之後 最下面的命令就是用來安裝的
第三步,打開Anaconda的prompt
就是Anaconda Prompt,打開之後就是命令行的形式
第四步。由於我們使用conda形式進行安裝,所以建議先更新一下conda
輸入命令 conda upgrade conda 然後一路確認 可能會更新很多東西,等就完事兒了
第五步,更換鏡像源,由於我們要下載的PyTorch的官方資源在國外,所以我們直接下載的話速度極慢而且會經常中斷,原因衆所周知,所以爲了解決這個問題,我們把源修改爲清華源
輸入命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
第六步,把剛纔在PyTorch官網的那條命令粘過來,這裏有個坑,大家千萬注意
原始命令後面有
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
一定要把 -c pytorch刪掉,因爲這一句的作用是指定了conda獲取pytorch的channel
,在此指定爲conda自帶的pytorch倉庫,不刪掉這個的話 我們的更新還是很慢,並且會經常由於網絡原因出錯中斷
第七步,我們在Pycharm中將py的解釋器選擇爲Anaconda,這樣我們就可以直接在pycharm中完成使用了
選擇之後,通過 import torch進行導入,不報錯就說明成功了!
上面的內容是我看了網上衆多大佬的總結並且自己踩坑無數之後得出的方法,希望能幫到需要的人
(寫代碼前的配置環境過程總是令人頭疼)