当 cuda 实际上是 12.0 时，pytorch for cuda 11.6 可以工作吗

Question

我将使用 pytorch 进行深度学习应用程序。我的计算机已经配备了 NVIDIA GPU。

我按照https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html安装了 CUDA

接下来，我使用 pytorch.org 上建议的 pip 命令在 conda 环境中安装了 pytorch

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

但是，当我检查我安装的 CUDA 版本时 nvcc - version，我得到

... Cuda compilation tools, release 12.0, V12.0.76 Build cuda_12.0.r12.0/compiler.31968024_0

当我运行代码时，我的安装似乎没问题 torch.cuda.is_available()

因为它返回 True 并正确识别 GPU 的名称。

只是想检查一下，当我实际输入具有大数据集的深度学习模型时，它是否万无一失，因为 CUDA 版本是 12.0，但我安装了适合 11.6 的 pytorch。

已经非常感谢了。

Answer 1

我实际上在 Pytorch 论坛上收到了以下针对同一问题的回复。

“是的，您的设置将起作用，因为 PyTorch 二进制文件附带了自己的 CUDA 运行时（以及其他 CUDA 库，例如 cuBLAS、cuDNN、NCCL 等）。本地安装的 CUDA 工具包（在您的情况下为 12.0）将仅是如果您从源代码或自定义 CUDA 扩展构建 PyTorch，则使用此选项。NVIDIA 驱动程序向后兼容，因此较新的驱动程序（我猜 525.60）也可以工作。”