详解Windows 11 上 CUDA 与 PyTorch 版本的兼容性

详解Windows 11 上CUDA 与 PyTorch 版本的兼容性

引言

在使用 NVIDIA GPU 的 Windows 11 系统上，理解 CUDA 版本和 PyTorch 的兼容性是至关重要的。nvidia-smi 命令提供了驱动支持的 CUDA 版本信息，但这与实际安装的 CUDA Toolkit 版本并不相同。本文将详细解释这两者的区别，并指导你如何检查和设置正确的 CUDA 版本以确保 PyTorch 能够正常运行。

nvidia-smi 显示的 CUDA 版本

当你在 Windows 11 上运行 nvidia-smi 命令时，如果看到类似以下输出：

CUDA Version: 12.6

这表示你的 NVIDIA 驱动程序支持 CUDA 12.6 版本。但请注意，这并不等同于你当前安装的 CUDA Toolkit 版本。

1. nvidia-smi 显示的 CUDA 版本

nvidia-smi 显示的 CUDA 版本是指你的 GPU 驱动程序支持的最高 CUDA 版本。即使你的驱动支持 CUDA 12.6，你可能安装了一个较低版本的 CUDA Toolkit（例如 CUDA 11.8），这两者是相互独立的。

2. 查看实际安装的 CUDA Toolkit 版本

要查看实际安装的 CUDA Toolkit 版本，你需要运行以下命令：

nvcc --version

这将显示你当前系统上安装的 CUDA 编译器（nvcc）的版本信息，例如：