linux安装pytorch-gpu

在线

首先确认你有显卡，并安装了合适的CUDA驱动，方法见此。

然后在此选择自己的安装平台、安装方式、语言等，如下图

得到安装命令，如

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

然后就可以安装了。如果顺利的话到这就结束了。

但是很可能很慢，因为-c pytorch代表的源在国外。因此如果你添加了国内的镜像，可以去掉这一句，直接用

conda install pytorch torchvision torchaudio cudatoolkit=10.2

离线

如果还是很慢，可以考虑手动下载（主要是pytorch的包很大，七百多兆，其他几个很小，在线装就行），离线安装。

利用上述命令提示的信息（运行之，查看信息，但拒绝其下载安装的请求），如

得到自己需要的安装包信息，然后搜索下载。如上图中显示我本来要下载的安装包是

pytorch-1.9.0-py3.8_cuda10.2_cudnn7.6.5_0

可以在各镜像源如此处找到此安装包。

下载后进入到安装包存放的目录，使用如下命令安装：

conda install --offline pytorch-1.9.0-py3.8_cuda10.2_cudnn7.6.5_0.tar.bz2

如果torch和torchvision都是手动安装，注意它们的版本要匹配：

详见此处。

检验

在python中运行

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 得到True

pip安装

以上是用conda安装，实际上在国内好像用pip装更快速方便些，直接

pip install torchvision

找出来就是gpu版，并附带找到对应的pytorch。