Linux下conda配置虚拟环境:python + pytorch
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下载皆可