pytorch安装

最新推荐文章于 2025-10-08 00:34:24 发布

原创最新推荐文章于 2025-10-08 00:34:24 发布 · 1.2k 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

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

安装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的安装的源变成了清华源，也就快了很多