本文主要介紹了Win10操作系統(tǒng)中PyTorch虛擬環(huán)境配置+PyCharm配置，分享給大家，具體如下：

注意版本，小可愛們?。?！

想直接看安裝方法的同學(xué)，可以看右邊欄的目錄，快速轉(zhuǎn)到哦。

試錯(cuò)經(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多兆的包要下半個(gè)小時(shí)左右，而且時(shí)間長可能會(huì)出現(xiàn)超時(shí)的情況：

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

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

此后終于走上了正途。

安裝版本

Python3.6

conda4.3.8

torch0.4.0

torchversion0.4.1

安裝過程

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

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

conda create -n pytorch-cpu python=3.6

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

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

conda info --envs

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

3，激活環(huán)境

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

activate pytorch36

4，升級(jí)pip

不出意外，這里的pip的版本會(huì)比較低，所以避免以后出錯(cuò)，還是提前升級(jí)一下省心

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)境pytorch36應(yīng)用到PyCharm工程中

如果show all里面沒有，點(diǎn)擊右邊的+號(hào)，手動(dòng)添加

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

選中之后，一路OK即可，稍等一會(huì)，就可以看到這里出現(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)

運(yùn)行結(jié)果

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

最新評論