PyTorch和CUDA版本对应关系

佰宝

已于 2024-03-01 09:14:16 修改

阅读量3.4w

点赞数 12

CC 4.0 BY-SA版权

分类专栏： # 深度学习文章标签： pytorch 深度学习人工智能

于 2022-11-02 17:28:39 首次发布

原文链接：https://www.cnblogs.com/zhoujiayingvana/p/15827369.html

深度学习专栏收录该内容

8 篇文章

订阅专栏

本文提供了PyTorch不同版本与CUDA版本之间的兼容性对照表，详细介绍了如何选择匹配的PyTorch和CUDA版本组合，以确保深度学习项目的顺利进行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

PyTorch和CUDA版本对应关系
截至2022.8.19
结论：10.2和11.3能兼容大部分版本的pytorch
官网链接：https://pytorch.org/get-started/previous-versions/
注意：注意低版本的 pytorch 是否支持更高版本的 cuda。（高版本的pytorch一般能兼容低版本cuda）
例如：你需要 1.7.0 的 pytorch，那么 cuda 只能 11.0 及以下。官方推荐的cuda版本为10.2和11.3，这两种 cuda 支持大多数的 pytorch 版本。
切换CUDA版本可参考这里

PyTorch 版本	CUDA 环境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	9.2
1.2.0、1.1.0、1.0.0(1)	10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	10.2
1.7.0(1)	11.0
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0	11.1
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	11.3
1.11.0	11.5
1.12.0(1)	11.6