PyTorch需哪版CUDA Toolkit - GPU

PyTorch 对 CUDA Toolkit 的版本要求取决于你安装的 PyTorch 版本，不是固定的。下面给你一个常见版本对照表（以 Linux x86_64 为主，Windows 类似）：

常见 PyTorch 与 CUDA 对应关系

PyTorch 版本	需要的 CUDA Toolkit
PyTorch 2.2 / 2.3	CUDA 11.8 或 12.1
PyTorch 2.1	CUDA 11.8 或 12.1
PyTorch 2.0	CUDA 11.7 或 11.8
PyTorch 1.13	CUDA 11.6 或 11.7
PyTorch 1.12	CUDA 11.6
PyTorch 1.11	CUDA 11.3

✅ 重要说明：
你不需要单独安装完整 CUDA Toolkit 也能用 GPU，只要装了对应 NVIDIA 驱动 + PyTorch 自带的 CUDA 运行时（cuDNN） 即可
如果你要自己编译 PyTorch / 写 CUDA 扩展，才需要安装对应版本的 CUDA Toolkit

如何查看你当前 PyTorch 用的 CUDA

import torch
print(torch.version.cuda)        # PyTorch 使用的 CUDA 版本
print(torch.cuda.is_available()) # 是否能用 GPU

建议

普通用户：直接用官方命令安装，例如

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

开发者：CUDA Toolkit 版本 ≥ PyTorch 要求的 CUDA 版本即可（向下不兼容）

如果你告诉我：

你的 PyTorch 版本
或 显卡驱动版本
或 操作系统

我可以给你精确推荐。