pytorch1.7可以和cuda12.1兼容吗
时间: 2024-02-22 09:01:10 浏览: 313
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
是的,PyTorch 1.7 可以和 CUDA 12.1 兼容。实际上,PyTorch 1.7 是专门为 CUDA 11.0 和 11.1 设计的,但它也可以与 CUDA 12.0 和 12.1 兼容。因此,如果你想在 PyTorch 中使用 CUDA 12.1,你可以安装 PyTorch 1.7,并且确保你的系统上已经安装了 CUDA 12.1 和相应的 cuDNN 版本。
阅读全文