下载网站：
https://pytorch.org/get-started/locally/
安装（不支持arm架构）
1.进入上方网站选择对应版本
2.直接在命令行输入命令行执行就可以
ARM架构安装
检查cuda是否可用
>>> import torch >>> torch.cuda.is_available() # 检查cuda是否可用 >>> torch.version.cuda # 查看cuda版本
检查cudnn是否可用
>>> torch.backends.cudnn.is_available() # 检查cudnn是否可用 >>> torch.backends.cudnn.version() # 查看cudnn版本
其他方面检测
torch.version.cuda()
会输出当前CUDA的版本，一般来说，会有 11.7 和 11.3
torch.version
输出当前pytorch的版本
torch.cuda.is_available()
查看cuda是否可用
torch.cuda.device_count()
查看显卡数量
torch.cuda.getdevicecapability()
获取GPU算力
