Windows PyTorch安装

GPU 驱动

先确保安装了硬件驱动

https://developer.nvidia.com/cuda/wsl
https://www.nvidia.com/en-us/drivers/

现在 WSL 的驱动 包含在 Windows 驱动中，都是同一驱动。

CUDA Toolkit

查看自己的GPU信息

nvidia-smi

这上面的 CUDA Version 表示此驱动支持的最高 CUDA版本，实际安装的 CUDA Toolkit 版本可低于此。

到官网下载低于 CUDA Version 的 CUDA Toolkit（太新了的记得看眼Pytorch有无适配）

https://developer.nvidia.com/cuda-toolkit
https://developer.nvidia.com/cuda-toolkit-archive 版本归档

像这个 13.1 就还没对应 pytorch

Pytorch

pytorch 各版本与 CUDA 和系统有严格的对应关系，请去官网查看命令

https://pytorch.org/get-started/previous-versions/

与 Python 的版本对应关系

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

PyTorch 版本	支持的 Python 版本	支持的 CUDA 版本
2.5	3.8, 3.9, 3.10, 3.11	11.7, 11.8, CPU
2.4	3.8, 3.9, 3.10	11.3, 11.6, CPU
2.3	3.7, 3.8, 3.9, 3.10	10.2, 11.1, CPU
2.2	3.7, 3.8, 3.9	10.2, 11.1, CPU
1.13	3.7, 3.8, 3.9, 3.10	10.2, 11.3, CPU
1.12	3.7, 3.8, 3.9, 3.10	10.2, 11.3, CPU