安装pytorch搭配cuda使用

问题

深度学习程序，在服务器运行，需要借助GPU加速。为了检测是否开启了GPU加速，采用以下代码：

~python
>> import torch
>> torch.cuda.is_available()
>> false  #说明没有使用GPU加速

安装过程

安装老版本的pytorch官方网站

搭配CUDA 11.1版本

命令行统一安装（网速限制，不建议使用）

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

离线下载进行安装（对应上面命令，一共三个whl后缀文件）

去官网下载适配本机的cuda=11.1的torch(https://download.pytorch.org/whl/torch_stable.html)
torch==1.9.0+cu111：

解释名词含义：
cu111: 表示cuda版本为11.1
torch-1.9.0：:表示torch版本为1.9.0
**%2B：表示命令里面的 + 号
cp37: 表示使用python版本为3.7
linux:表示适用于linux系统
x86_64:表示同时兼容32和64位系统

torchvision==0.10.0+cu111：

torchaudio==0.9.0：

最后安装三个whl后缀的文件

pip install 下载的文件所在目录/文件名

可以检验一下是否安装成功，进入python环境，输入如下命令：

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

如果输出是true，就代表安装成功。