Linux ubuntu安装pytorch

0、前置工作

在Anaconda中创建pytorch环境

conda create -n pytorch python=3.8 
# 在这之前先使用 python --version 查看python的版本号

1、激活环境

source activate pytorch

一、直接安装

打开pytorch官网：https://pytorch.org/
选择需要版本：

复制Command指令到命令行，执行

等待安装完成即可
缺点：安装很慢，有时候中间还会断。（windows还能科学上网，ubuntu上就不太会了。。。）

二、国内镜像安装

2.1、安装

执行 ls -a，可以看到有 .condarc 文件，可以直接进入进行镜像通道添加，或者如下执行指令
在命令行执行：

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/conda-forge/
conda config --set show_channel_urls yes

2.2、问题

但是接下来执行上述安装 pytorch 指令，会报 Http 错误，无法进行下载。

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.tuna.tsi...

在网上查了一下，有以下解决方案：

• 打开 .condarc 文件，删除channels中的 -defaults
• 若上述仍无效，将前面添加的清华镜像源中的 https 修改为 http
• 再执行 pytorch 的conda安装指令。
   本人到这一步确实可以运行了，但我明明安装的应该是带有 cuda 的pytorch版本，之后在 python 中使用 torch.cuda.is_availabel() 返回却是false。
   经过排查，似乎是清华镜像源中并没有该版本的cuda pytorch，所以就下载的cpu版本 pytorch。

三、本地安装

1、进入Anaconda资源包下载链接
 https://conda.anaconda.org/pytorch/linux-64/

2、找到需要的资源包

可以直接通过浏览器下载，个人感觉用迅雷下载比较快。
 3、复制到服务器 ~/anaconda3/pkgs 文件夹下

4、执行本地安装指令

conda install --use-local ./anaconda3/pkgs/pytorch-1.10.0-py3.8_cuda10.2_cudnn7.6.5_0.tar.bz2

这里可能会需要下载其他的依赖包，但相较于700多M的pytorch包来说已经小很多了。

5、测试

import torch
torch.cuda.is_available()

若返回True则表示安装成功。