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

本文主要介紹瞭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)

運行結果

到此這篇關於Win10操作系統中PyTorch虛擬環境配置+PyCharm配置的文章就介紹到這瞭,更多相關PyTorch虛擬環境配置+PyCharm配置內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: