本文主要介紹了Win10操作系統(tǒng)中PyTorch虛擬環(huán)境配置+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，創(chuàng)建pytorch36虛擬環(huán)境（因為python是3.6所以這里記為36）

pytorch-cpu可以自己設名字，這里我用的是pytorch36

conda create -n pytorch-cpu python=3.6

這樣就算創(chuàng)建成功了

2，查看自己曾經創(chuàng)建的虛擬環(huán)境

conda info --envs

 帶星號*的就是當前激活的環(huán)境

3，激活環(huán)境

注意以后的操作均是在pytorch36虛擬環(huán)境中操作！

activate pytorch36

4，升級pip

不出意外，這里的pip的版本會比較低，所以避免以后出錯，還是提前升級一下省心

python -m pip install --upgrade pip

5，安裝torch-0.4.0（CPU版）

查閱網上資料，發(fā)現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，將虛擬環(huán)境pytorch36應用到PyCharm工程中

如果show all里面沒有，點擊右邊的+號，手動添加

找到自己虛擬環(huán)境的安裝目錄（找不到的話，可以直接在我的電腦里搜你創(chuàng)建的環(huán)境名稱）

選中之后，一路OK即可，稍等一會，就可以看到這里出現了原先創(chuàng)建的環(huán)境

選中OK/Apply即可

9，測試

創(chuàng)建test.py文件

import torch
print(torch)
x1 = torch.Tensor([3, 4])
x2 = torch.FloatTensor([1, 2])
print(x1)
print(x2)

運行結果

到此這篇關于Win10操作系統(tǒng)中PyTorch虛擬環(huán)境配置+PyCharm配置的文章就介紹到這了,更多相關PyTorch虛擬環(huán)境配置+PyCharm配置內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

最新評論