cuda11.4对应python和pytorch版本
时间: 2024-04-15 09:14:23 浏览: 362
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
CUDA 11.4 目前支持的 Python 版本是 3.6、3.7、3.8 和 3.9,PyTorch 的官方文档中推荐使用 CUDA 11.1 或 11.2,但也支持使用 CUDA 11.4。因此,可以使用 Python 3.6-3.9 和 PyTorch 1.9 来配合 CUDA 11.4 使用。
阅读全文