【python-pytorch与cuda版本对应关系汇总】此文章归类为:["python","机器学习","深度学习","pytorch","人工智能"]。
- 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 |
更多【python-pytorch与cuda版本对应关系汇总】相关视频教程:www.yxfzedu.com