CUDA+PyTorch安装

CUDA+PyTorch安装

一、版本匹配

1.1 查看显卡驱动版本

nvidia-smi

显卡驱动版本为551.86，最高支持CUDA版本为12.4

若要更新显卡驱动，可以使用 NVIDIA GeForce Experience 官方程序进行更新

1.2 查看显卡驱动版本和CUDA版本匹配

链接: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
从官网中的表3可以看到当前显卡驱动已支持所有版本CUDA

1.3 查看PyTorch版本

链接: https://pytorch.org/get-started/locally/

根据实际需求选择，只需要显卡驱动满足对应CUDA版本要求即可，这里选择CUDA12.1计算平台

二、安装CUDA

2.1 下载CUDA

步骤一已确定CUDA版本不低于12.1，这里就选择安装CUDA 12.1版本，当然也可选择更高版本向后兼容

官网下载页默认为最新版本，安装12.1版本在搜索栏输入12.1搜索即可

可以看到12.1有两个版本，这里选择较新版本【12.1 Update 1】

根据实际需求选择好相应平台安装版本即可，安装方式分为本地安装【exe (local)】和网络安装【exe (network)】两种，推荐网络安装，下载速度很快

2.2 安装CUDA

运行安装程序，弹出的文件选择框为临时文件解压地址，安装完成后会自动删除，默认即可

进入安装界面后选择精简安装即可（缺图）

安装验证

nvcc -V

如图即为安装成功

2.3 安装PyTorch

步骤1.3已选择安装版本，直接运行官网给出的安装命令即可

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

验证PyTorch

import torch
x = torch.rand(5, 3)
print(x)

验证CUDA调用

import torch
torch.cuda.is_available()