ubuntu下conda pytorch1.3_gpu安装方法
新建环境,避免与基础环境出现冲突
conda create -n pytorch python=3.7
切换到pytorch环境
conda activate pytorch
添加清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
另外为了保险起见,建议同时添加第三方conda源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
安装PyTorch
conda install pytorch torchvision cudatoolkit=10.0
退出虚拟环境
source deactivate pytorch
删除环境,删除之前必须退出虚拟环境
conda remove -n pytorch --all