1:下载cuda工具cuda-toolkit
地址:https://developer.nvidia.com/cuda-toolkit-archive
2:一路向下:安装
3:配置环境变量
path 环境变量中添加:
4:验证cuda是否安装成功:
nvcc -V
set cuda
5:安装cuDNN
登录cuDNN下载网页(必须注册账号并登录)并安装
cuDNN Archive | NVIDIA Developer
将解压后的文件夹中的bin、include、lib 三个文件夹,移动到CUDA Development 安装路径下,
与同名文件夹合并。
6:安装对应pytorch版本:
Start Locally | PyTorch
选择对应的版本,安装:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
6:验证cuda是否成功:
import torch
print(torch.cuda.is_available())