注意版本,小可愛們!!!
想直接看安裝方法的同學,可以看右邊欄的目錄,快速轉到哦。
試錯經歷
一開始試着裝1.0版本的PyTorch:
pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl
然而在國內下載這些非常慢,70多兆的包要下半個小時左右,而且時間長可能會出現超時的情況:
有同學使用清華源/科大源來加速,但是我試了一下仍然不行o( ̄┰ ̄*)ゞ,最後還是採用這種方式。經歷過三次超時,終於。。。(原來超時也是碰運氣的)
然而安裝torchvision時爆出了錯誤,,,我裝了一下午臉終於白了一次,你給我說沒有匹配的版本o( ̄┰ ̄*)ゞ
此後終於走上了正途。
安裝版本
Python3.6
conda4.3.8
torch0.4.0
torchversion0.4.1
安裝過程
1,創建pytorch36虛擬環境(因爲python是3.6所以這裏記爲36)
pytorch-cpu可以自己設名字,這裏我用的是pytorch36
conda create -n pytorch-cpu python=3.6
這樣就算創建成功了
2,查看自己曾經創建的虛擬環境
conda info --envs
帶星號*的就是當前激活的環境
3,激活環境
注意以後的操作均是在pytorch36虛擬環境中操作!
activate pytorch36
4,升級pip
不出意外,這裏的pip的版本會比較低,所以避免以後出錯,還是提前升級一下省心
python -m pip install --upgrade pip
5,安裝torch-0.4.0(CPU版)
查閱網上資料,發現0.4版本已經可以滿足大部分需求,而且GitHub上大多代碼是依託於0.4版本的torch
pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
可以看出早期的版本體積較小,即使是從原網站下載,速度也還是能接受的
6,安裝torchvision0.4.1
pip install torchvision==0.4.1
7,測試是否安裝成功
import torch
print(torch)
如下,可以成功導入torch,並輸出<module .....>就說明安裝成功了
8,將虛擬環境pytorch36應用到PyCharm工程中
如果show all裏面沒有,點擊右邊的+號,手動添加
找到自己虛擬環境的安裝目錄(找不到的話,可以直接在我的電腦裏搜你創建的環境名稱)
選中之後,一路OK即可,稍等一會,就可以看到這裏出現了原先創建的環境
選中OK/Apply即可
9,測試
創建test.py文件
import torch
print(torch)
x1 = torch.Tensor([3, 4])
x2 = torch.FloatTensor([1, 2])
print(x1)
print(x2)
運行結果
完結撒花*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。