Win10操作系統(tǒng)中PyTorch虛擬環(huán)境配置和PyCharm配置的示例分析

小編給大家分享一下Win10操作系統(tǒng)中PyTorch虛擬環(huán)境配置和PyCharm配置的示例分析，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

具體如下：

試錯經(jīng)歷

一開始試著裝1.0版本的PyTorch：

pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl

然而在國內(nèi)下載這些非常慢，70多兆的包要下半個小時左右，而且時間長可能會出現(xiàn)超時的情況：

有同學(xué)使用清華源/科大源來加速，但是我試了一下仍然不行o(￣┰￣*)ゞ，最后還是采用這種方式。經(jīng)歷過三次超時，終于。。。（原來超時也是碰運氣的）

然而安裝torchvision時爆出了錯誤，，，我裝了一下午臉終于白了一次，你給我說沒有匹配的版本o(￣┰￣*)ゞ

此后終于走上了正途。

安裝版本

Python3.6

conda4.3.8

torch0.4.0

torchversion0.4.1

安裝過程

1，創(chuàng)建pytorch46虛擬環(huán)境（因為python是3.6所以這里記為36）

pytorch-cpu可以自己設(shè)名字，這里我用的是pytorch46

conda create -n pytorch-cpu python=3.6

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

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

conda info --envs

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

3，激活環(huán)境

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

activate pytorch46

4，升級pip

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

python -m pip install --upgrade pip

5，安裝torch-0.4.0（CPU版）

查閱網(wǎng)上資料，發(fā)現(xiàn)0.4版本已經(jīng)可以滿足大部分需求，而且GitHub上大多代碼是依托于0.4版本的torch

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl

可以看出早期的版本體積較小，即使是從原網(wǎng)站下載，速度也還是能接受的

6，安裝torchvision0.4.1

pip install torchvision==0.4.1

7，測試是否安裝成功

import torch
print(torch)

 如下，可以成功導(dǎo)入torch，并輸出<module .....>就說明安裝成功了

8，將虛擬環(huán)境pytorch46應(yīng)用到PyCharm工程中

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

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

選中之后，一路OK即可，稍等一會，就可以看到這里出現(xiàn)了原先創(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)

運行結(jié)果

以上是“Win10操作系統(tǒng)中PyTorch虛擬環(huán)境配置和PyCharm配置的示例分析”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！