pytorch安装

最新推荐文章于 2025-07-09 15:04:37 发布

maohule

最新推荐文章于 2025-07-09 15:04:37 发布

阅读量1.1k

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/maohule/article/details/90694540

本文介绍了在Ubuntu系统下安装Pytorch的方法。安装前需查看cuda版本号，若安装速度慢，可去掉 -c pytorch并修改为清华源。还给出指定python版本的安装命令，以及清华源添加方法，替换源后可提升安装速度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装pytorch之前先查看ubuntu下cuda的版本号，再进行安装

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

如果安装速度慢的话，去掉-c pytorch。因此“-c pytorch”默认从pytorch官网下载安装包，因此安装的时候将-c pytorch去掉，再修改为清华源，可提高安装速度。

指定python版本

conda install python=2.7 pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

基于cuda10.1版本的安装

conda install python=2.7 pytorch=1.3.0 torchvision cudatoolkit=10.1

PS：

清华源添加方法：

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

然后系统会自动在/home中增加.condarc文件，包括如下信息：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - default
show_channel_urls: true

有的简称写到，需要将 - default 那一行删掉，我没有删掉，目前没有什么影响。接着保存.condarc .然后重启终端就能用conda安装新的库了。

替换了源之后，pytorch的安装的源变成了清华源，也就快了很多