pytorch安裝

注：在訓(xùn)練模型的時候，有時候可能需要不同版本的 torch和torchvision，所以需要配置不同的環(huán)境。anconda和pycharm自行安裝，接下來在pycharm終端pip安裝。

1. torch和torchvision下載

進入pytorch官網(wǎng)，[https://pytorch.org]

進入右下角的網(wǎng)站下載，找到需要的版本，我的版本如下

1.3.0-版本 cp37-python版本3.7 win-Windows系統(tǒng)

2. pycharm終端安裝

（1）首先創(chuàng)建一個環(huán)境

conda create -n pytorch_1.3 python=3.7

注：有可能出現(xiàn)這個error：You will need to adjust your conda configuration to proceed. Use `conda config --show channels 原因：鏡像停止服務(wù)了，方法：只要刪除所有鏡像鏈接，恢復(fù)到默認(rèn)設(shè)置即可 conda config --remove-key channels 輸入conda config --show channels 可以查看你的channels

（2）中間會出現(xiàn)一個判斷，輸入Y ，這樣你的環(huán)境創(chuàng)建好了，然后安裝你的torch和torchvision 通過cd命令進入你下載的環(huán)境當(dāng)中 dir` 可以查看你的該路徑下的文件

（3）torch安裝

pip install “torch-1.3.0+cpu-cp37-cp37m-win_amd64.whl”`

雙引號中的可在dir下復(fù)制，因為安裝中有一個numpy下載，速度比較慢建議使用清華源下載

pip install -i 清華鏡像網(wǎng)址（http） “torch-1.3.0+cpu-cp37-cp37m-win_amd64.whl”

（4）torchvision安裝同上

pip install "torchvision-0.4.0+cu92-cp37-cp37m-win_amd64.whl"

注：有可能出現(xiàn)error timed out 原因：網(wǎng)速過慢方法：這個可以解決類似安裝一些package可用

pip --default- timeout=100 intall "torchvision-0.4.0+cu92-cp37-cp37m-win_amd64.whl"

(5)終端查看

pip list

可以查看安裝的版本，最后在pycharm的解釋器下配置你的環(huán)境

3.環(huán)境切換

復(fù)現(xiàn)有些模型可能需要不同的torch和torchvision版本，需要創(chuàng)建新的環(huán)境進行切換，終端下輸入 conda deactivate 可推出當(dāng)前的環(huán)境，接著可以按照一開始的步驟創(chuàng)建你的環(huán)境，在(2)后輸入 pip activate pytorch_1.3 這樣就進入你創(chuàng)建的環(huán)境當(dāng)中，接著可以安裝你需要的版本。

總結(jié)

到此這篇關(guān)于Pycharm中切換pytorch的環(huán)境和配置的文章就介紹到這了,更多相關(guān)Pycharm中切換pytorch的環(huán)境和配置內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論