CUDA迷思｜ torch.version.cuda； nvcc -V； nvidia-smi分别是什么

torch.version.cuda

print(f"torch.__version__: {torch.__version__}")

# Output
# torch.version.cuda: 10.2

torch.version.cuda 只是定义为字符串。它不查询任何内容，不会告诉您安装了哪个版本的 CUDA。它仅告诉您安装的 PyTorch 适用于该 (10.2) 版本的 CUDA。

nvcc -V

nvcc -V

# Output
# Cuda compilation tools, release 11.8, V11.8.89

但您系统上实际运行的 CUDA 版本是 11.8。

nvidia-smi

nvidia-smi 的输出是显卡的硬件信息，它代表了Driver CUDA版本，与框架程序的Runtime CUDA并不相同。可以理解为能够匹配的最高CUDA版本。