一般来说Python3.x对应着PyTorch1.x。

以下是一些常见的Python和PyTorch版本对应关系。请注意,随着时间的推移,版本可能会有所改变,因此在使用时请务必查阅最新文档以获取准确的信息。

PyTorch 版本 对应的 Python 版本
PyTorch 1.0 Python 2.7, 3.5, 3.6, 3.7
PyTorch 1.1 Python 2.7, 3.5, 3.6, 3.7
PyTorch 1.2 Python 2.7, 3.5, 3.6, 3.7
PyTorch 1.3 Python 2.7, 3.5, 3.6, 3.7
PyTorch 1.4 Python 2.7, 3.5, 3.6, 3.7, 3.8
PyTorch 1.5 Python 3.5, 3.6, 3.7, 3.8
PyTorch 1.6 Python 3.5, 3.6, 3.7, 3.8
PyTorch 1.7 Python 3.5, 3.6, 3.7, 3.8
PyTorch 1.8 Python 3.6, 3.7, 3.8, 3.9
PyTorch 1.9 Python 3.6, 3.7, 3.8, 3.9

请注意,Python 2.x 分支在2020年已经停止维护,因此强烈建议使用Python 3.x 版本。

加一张附表

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
Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐