使用pip离线安装GPU版本的Pytorch

安装CUDA和cuDNN

两者之间的依赖关系请看这儿。

检查显卡的驱动程序与cuda版本是否协调

CUDA对英伟达显卡驱动的版本号有一个最低要求，不满足的话，后续再如何配置也无法使用GPU版的pytorch,依赖关系看这儿。如果不满足要求，可以从这个网址下载显卡的驱动程序。

pip离线下载

首先配置pip的源为清华源，然后使用如下命令下载(win10系统),

pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

清华源：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

pip离线安装

使用命令pip install --no-index --find-links . 安装包名称安装，其中.表示pytorch安装包的位置在当前目录中。

验证方式

import torch
print(torch.cuda.is_available())