pytorch与cuda版本对应关系汇总
- 开发
- 30
-
- pytorch与cuda版本关系
cuda版本 |
支持pytorch版本 |
cuda10.2 |
1.5 ~ 1.12 |
cuda11.0 |
1.7 ~ 1.7.1 |
cuda11.1 |
1.8 ~ 1.10.1 |
cuda11.3 |
1.8.1 ~ 1.12.1 |
cuda11.6 |
1.12.0 ~ 1.13.1 |
cuda11.7 |
1.13.0 ~ 2.0.1 |
cuda11.8 |
2.0.0 ~ 2.1.1 |
cuda12.1 |
2.1.0 ~ 2.1.1 |
- cuda 与 cudnn关系
cuda版本 |
支持cudnn版本 |
cuda10.2 |
v7.6.5 ~ v8.7.0 |
cuda11.3 |
v8.2.0 ~ v8.9.6 |
- pytorch 与 python关系
torch |
torchvision |
Python |
main / nightly |
main / nightly |
>=3.8, <=3.11 |
2.0 |
0.15 |
>=3.8, <=3.11 |
1.13 |
0.14 |
>=3.7.2, <=3.10 |
1.12 |
0.13 |
>=3.7, <=3.10 |
1.11 |
0.12 |
>=3.7, <=3.10 |
1.10 |
0.11 |
>=3.6, <=3.9 |
1.9 |
0.10 |
>=3.6, <=3.9 |
1.8 |
0.9 |
>=3.6, <=3.9 |
1.7 |
0.8 |
>=3.6, <=3.9 |
1.6 |
0.7 |
>=3.6, <=3.8 |
1.5 |
0.6 |
>=3.5, <=3.8 |
1.4 |
0.5 |
==2.7, >=3.5, <=3.8 |
1.3 |
0.4.2 / 0.4.3 |
==2.7, >=3.5, <=3.7 |
1.2 |
0.4.1 |
==2.7, >=3.5, <=3.7 |
1.1 |
0.3 |
==2.7, >=3.5, <=3.7 |
<=1.0 |
0.2 |
==2.7, >=3.5, <=3.7 |
原文地址:https://blog.csdn.net/u011489887/article/details/135250561
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。
本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若转载,请注明出处:https://www.suanlizi.com/kf/1740233638123016192.html
如若内容造成侵权/违法违规/事实不符,请联系《酸梨子》网邮箱:1419361763@qq.com进行投诉反馈,一经查实,立即删除!