windows配置cuda+pytorch

1.nvidia driver + cuda

下载版本匹配的driver和cuda toolkit并安装，精简安装，完全按照指示安装就行。

查看driver和cuda对应关系：

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

检查安装是否成功，打开cmd执行：

nvidia-smi

nvcc -V

两个命令都成功输出，那就安装成功了。

2.pytorch

默认安装的是cpu支持版本的。所以先卸载之前安装的pytorch

pip uninstall torch torchvision torchaudio

安装cuda支持版本的pytorch,跟cuda有对应关系,比如cuda12.4要指定"cu124"

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

cuda12.8指定"cu128"

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128