查看位置对照
| 命令 | 显示内容 | 含义 |
|---|---|---|
| nvidia-smi | CUDA Version: xx.y | 该驱动支持到的 CUDA 运行时上限 |
| nvcc --version | release xx.y | 本机安装的 CUDA 工具箱版本 |
| python -c torch… | torch.version.cuda | PyTorch 编译时使用的 CUDA 版本 |
三步核对
三步核对
nvidia-smi | head -5
nvcc --version
python -c "import torch; print(torch.version.cuda)"容器内看到的 Toolkit 版本来自镜像，与宿主机 nvcc 无关；判断容器能否用 GPU 要看驱动与容器 CUDA 运行时的兼容关系。