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
posted @   有何m不可  阅读(2660)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· .NET10 - 预览版1新功能体验(一)
点击右上角即可分享
微信分享提示