天天看點

Ubuntu 18.04 安裝 Anaconda 配置 pytorch 虛拟環境1、安裝 Anaconda 2、配置pytorch虛拟環境3、測試pytorch

此為記錄在伺服器上配置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的路徑
           

繼續閱讀