ubuntu 使用国内源来安装/卸载 pytorch

最新推荐文章于 2024-07-22 11:08:20 发布

梦dancing

最新推荐文章于 2024-07-22 11:08:20 发布

阅读量3.1k

点赞数

分类专栏： Ubuntu+python

原文链接：https://blog.csdn.net/qq_29462849/article/details/90545782

版权

Ubuntu+python 专栏收录该内容

30 篇文章 12 订阅

订阅专栏

一、找到合适的安装方式

pytorch官网：https://pytorch.org/

å¨è¿éæå¥å¾çæè¿°

二、安装命令

pip --default-timeout=100 install torch torchvision -i  https://pypi.tuna.tsinghua.edu.cn/simple

或者

conda install pytorch torchvision -c https://pypi.mirrors.ustc.edu.cn/simple

后面的是镜像，当然也可以采用其他源，网上很多，在这里不再陈述。或者用以下命令

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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/pytorch/
conda config --set show_channel_urls yes

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0