Win10操作系統:PyTorch虛擬環境配置+PyCharm配置

注意版本,小可愛們!!!

想直接看安裝方法的同學,可以看右邊欄的目錄,快速轉到哦。

試錯經歷

一開始試着裝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)

運行結果

完結撒花*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。

 

 

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章