此為記錄在伺服器上配置pytorch環境步驟。
目錄
1、安裝 Anaconda
2、配置pytorch虛拟環境
1、使用activate激活該環境
2、安裝pytorch
3、測試pytorch
1、安裝 Anaconda
此次任務需要用 python 2,是以安裝 Anaconda 的 Python 2.7 版本。具體安裝過程與 之前安裝Anaconda 3 無異。
不過還是推薦在官網上 https://www.anaconda.com/download/#linux 右擊 Python 2.7 version 下的 Download (Linux 64位版本), 選擇複制連結位址,擷取下載下傳連結後,線上安裝,在終端輸入:
wget https://repo.anaconda.com/archive/Anaconda2-2019.07-Linux-x86_64.sh
之後安裝過程,請參見 之前安裝Anaconda 3 。
2、配置pytorch虛拟環境
conda create -n pytorch python=2.7
1、使用activate激活該環境
source activate pytorch
或
conda activate pytorch
激活後,會看到前面會顯示環境名字比如(pytorch)。可确認python和pip的路徑是否在envs下面:
which python
which pip
[Update(2020-12-10):
(1) 删除該環境
conda remove -n pytorch --all
(2) 檢視環境資訊, 帶星号*的是目前的環境
conda info --env
(Windows裡面可以:conda-env list )
(3) Windows 中激活或退出環境
激活:activate pytorch
退出:deactivate
(比Ubuntu簡略一點)
Done.]
2、安裝pytorch
這是最關鍵的一步,有人推薦在這裡指定固定的版本,如cuda8.0、python3.6對應版本:conda install pytorch torchvision cuda80 -c pytorch 。我依稀記得伺服器上cuda10.0,但又不想費時間去确定,随将這個問題扔給了conda,讓它自己選擇。
conda install pytorch torchvision
若報錯或失敗,則可
conda install pytorch torchvision -c pytorch
最後的-c pytorch參數的意思是強制在pytorch官網下載下傳。有時候因為牆的緣故,可能會因為這一參數導緻下載下傳失敗。
(torchvision 很有用,是以,将 torchvision 和 pytorch 一同安裝。)
也有用pip或源碼安裝的,但那樣更麻煩,随棄。
3、測試pytorch
在終端打開 python 後:
import torch
import torchvision
print(torch.__version__) #輸出PyTorch版本
print(torch.__path__) #輸出PyTorch的路徑