1. 程式人生 > 程式設計 >Pycharm中切換pytorch的環境和配置的教程詳解

Pycharm中切換pytorch的環境和配置的教程詳解

pytorch安裝

注:在訓練模型的時候,有時候可能需要不同版本的 torch和torchvision,所以需要配置不同的環境。anconda和pycharm自行安裝,接下來在pycharm終端pip安裝。

1. torch和torchvision下載

進入pytorch官網,[https://pytorch.org]

pip安裝,其他根據就需要直接選擇在這裡插入圖片描述

進入右下角的網站下載,找到需要的版本,我的版本如下

Pycharm中切換pytorch的環境和配置的教程詳解

1.3.0-版本 cp37-python版本3.7 win-Windows系統

在這裡插入圖片描述

2. pycharm終端安裝

(1)首先建立一個環境

conda create -n pytorch_1.3 python=3.7

Pycharm中切換pytorch的環境和配置的教程詳解

注:有可能出現這個error:You will need to adjust your conda configuration to proceed. Use `conda config --show channels 原因:映象停止服務了,方法:只要刪除所有映象連結,恢復到預設設定即可 conda config --remove-key channels 輸入conda config --show channels 可以檢視你的channels

(2)中間會出現一個判斷,輸入Y ,這樣你的環境建立好了,然後安裝你的torch和torchvision 通過cd命令進入你下載的環境當中 dir` 可以檢視你的該路徑下的檔案

在這裡插入圖片描述

(3)torch安裝

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

雙引號中的可在dir下複製,因為安裝中有一個numpy下載,速度比較慢建議使用清華源下載

pip install -i 清華映象網址(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"

注:有可能出現error timed out 原因:網速過慢方法:這個可以解決類似安裝一些package可用

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

在這裡插入圖片描述

(5)終端檢視

pip list

在這裡插入圖片描述

可以檢視安裝的版本,最後在pycharm的直譯器下配置你的環境

3.環境切換

復現有些模型可能需要不同的torch和torchvision版本,需要建立新的環境進行切換,終端下輸入 conda deactivate 可推出當前的環境,接著可以按照一開始的步驟建立你的環境,在(2)後輸入 pip activate pytorch_1.3

這樣就進入你建立的環境當中,接著可以安裝你需要的版本。

總結

到此這篇關於Pycharm中切換pytorch的環境和配置的文章就介紹到這了,更多相關Pycharm中切換pytorch的環境和配置內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!