Python、PyTorch与CUDA版本对应指南表

常见的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

作者:码韵

物联沃分享整理
物联沃-IOTWORD物联网 » Python、PyTorch与CUDA版本对应指南表

发表回复