ubuntu下pytorch1.3安装方法

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

猜你喜欢

转载自blog.csdn.net/qq_35911756/article/details/107391396
今日推荐