Python、PyTorch与cuda的版本对应表

码韵 2024-09-14 11:35:01 阅读 83

常见的Python和PyTorch版本对应关系

PyTorch版本 对应的Python
PyTorch1.0 Python 2.7,3.5,3.6,3.7
PyTorch1.1 Python 2.7,3.5,3.6,3.7
PyTorch1.2 Python 2.7,3.5,3.6,3.7
PyTorch1.3 Python 2.7,3.5,3.6,3.7
PyTorch1.4 Python 2.7,3.5,3.6,3.7,3.8
PyTorch1.5 Python 3.5,3.6,3.7,3.8
PyTorch1.6 Python 3.5,3.6,3.7,3.8
PyTorch1.7 Python 3.5,3.6,3.7,3.8
PyTorch1.8 Python 3.6,3.7,3.8,3.9
PyTorch1.9 Python 3.6,3.7,3.8,3.9

常见的Torch、CUDA和Python版本的对应关系

Torch版本

可选的CUDA版本

支持的Python版本

2.0.1 cu117,cu118 cp38,cp39,cp310,cp311
2.0.0 cu117,cu118 cp38,cp39,cp310,cp311
1.13.1 cu116,cu117 cp37,cp38,cp39,cp310
1.13.0 cu116,cu117 cp37,cp38,cp39,cp310
1.12.1 cu113,cu116 cp37,cp38,cp39,cp310
1.12.0 cu113,cu116 cp37,cp38,cp39,cp310
1.11.0 cu113,cu115 cp37,cp38,cp39,cp310
1.10.2 cu102,cu111,cu113 cp36,cp37,cp38,cp39
1.10.1 cu102,cu111,cu113 cp36,cp37,cp38,cp39
1.10.0 cu102,cu111,cu113 cp36,cp37,cp38,cp39
1.9.1 cu102,cu111 cp36,cp37,cp38,cp39
1.9.0 cu102,cu111 cp36,cp37,cp38,cp39
1.8.1 cu101,cu102,cu111 cp36,cp37,cp38,cp39
1.8.0 cu101,cu111 cp36,cp37,cp38,cp39
1.7.1 cu101,cu110 cp36,cp37,cp38,cp39
1.7.0 cu101,cu110 cp36,cp37,cp38
1.6.0 cu101 cp36,cp37,cp38
1.5.1 cu92,cu101 cp35,cp36,cp37,cp38
1.5.0 cu92,cu101 cp35,cp36,cp37,cp38
1.4.0 cu92 cp35,cp36,cp37,cp38
1.3.1 cu92 cp35,cp36,cp37
1.3.0 cu92 cp35,cp36,cp37
1.2.0 cu92 cp35,cp36,cp37


声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。