一、查看显卡型号信息

系统：Windows11
显卡：GeForce GTX 1060
型号：
（1）搜索 NVIDIA，选择 NVIDIA Control Panel

（2）打开 NVIDIA control Panel，打开系统信息，点击组件，找到 NVCUDA.DLL,查看版本号：11.1.106
说明我们可以下载 11.1.106 及其以下版本的 CUDA

二、下载 CUDA

https://developer.nvidia.com/cuda-toolkit-archive
我们选 CUDA Toolkit 11.1.0

选择 Windows => x86_64 => 10（Windows 10/11 可以用）=> exe (local) => Download

下载完成后，点击安装

下载 CUDNN

https://developer.nvidia.com/rdp/cudnn-download
选择 Windows => x86_64 => 11 => exec (local) => Download

下载完成后，点击安装

安装 Pytorch

pip install torch==1.8.0+cu111 \torchvision==0.9.0+cu111 \torchaudio==0.8.0 \-f https://download.pytorch.org/whl/torch_stable.html

测试验证

# Python 环境
>>> import torch
>>> torch.cuda.is_available()
True

参考

https://blog.csdn.net/weixin_45955630/article/details/113105064