Linux下conda配置虛擬環境:python + pytorch
默認已經安裝好conda
創建虛擬環境
conda創建並激活虛擬環境
命令:
conda create -n your_env_name python=2.7/3.6
source activate your_env_name
其中,-n中n表示name,即你創建環境的名字。
之后如果忘記自己創建的環境的名字,可以查看conda中的環境:
conda env list
之后選擇你創建的環境激活。之后進入pytorch官網選擇你要安裝的版本以及操作系統、CUDA版本號,便可以得到下載的命令。比如本人得到的命令是:
conda install pytorch torchvision cuda80 -c pytorch
若不知道自己CUDA的版本號,可以輸入下面命令查詢:
cat /usr/local/cuda/version.txt
若希望快速下載,可以把源換為國內的鏡像源,從而提高下載速度。在下載前輸入:
vim ~/.condarc
打開文件后,輸入:
channels:
- https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
- https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- defaults
show_channel_urls: true
若配置好環境后需要別的包,用conda或者pip下載皆可