一、cuda安装：

1. nvidia-smi来查看当前安装的驱动版本

显示：NVIDIA-SMI 512.72 Driver Version: 512.72 CUDA Version: 11.6

2. 根据显卡型号和驱动程序版本，到CUDA官方开发者网站(https://developer.nvidia.com/cuda-toolkit-archive)找到对应显卡支持的CUDA版本列表（找11.6版本）

二、GPU版pytorch安装

1. 安装pytorch包

# CUDA 11.6
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

2. 测试

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

输出：

1.13.1+cu116
True
1

[1] https://tongyi.aliyun.com/qianwen/share?shareId=5359f0b1-8e58-42d3-9bfd-89a7f2c8b6b9

[2] https://pytorch.org/get-started/previous-versions/