pytorch官网：pytorch.org/
查看自己的cuda版本
终端执行命令nvidia-smi
右上角即为你的cuda版本
按照官方的指示安装pythorch
笔者cuda版本是11.6，按照自己的需要下载相应版本
官网版本下载速度有点慢，指定清华源下载。笔者电脑有点老了，cuda版本是11.6，貌似目前都是11.8起步了，暂时下载cpu版本的了
设置清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
测试是否安装成功
import torch
print(torch.__version__)
torch.cuda.is_available()
要是装的是gpu版本，最后
torch.cuda.is_available()的返回值是true
