Home Arrow Icon Knowledge base Arrow Icon Global Arrow Icon python 3.11でpytorch/xlaを使用できますか


python 3.11でpytorch/xlaを使用できますか


現在、Pytorch/XLAは、LinuxディストリビューションのPythonバージョンを3.11までサポートしています。ただし、特定の要件と制限があります。たとえば、「TORCH_XLA`バージョン2.5をインストールするには、Python 3.12以降のホイールが利用できないため、Python 3.9から3.11との互換性を確保する必要があります[3] [4]。さらに、Pytorch自体はWindowsでPython 3.11をサポートしていません[7]。

Pytorch/XLAバージョン2.6以降の場合、特定のコマンドを使用してインストールできるC ++ 11 ABIで構築されたホイールを含むPython 3.11のサポートがあります[5]。したがって、Pytorch/XLAはLinuxでPython 3.11で使用できますが、慎重なバージョン管理とインストールが必要です。

引用:
[1] https://github.com/pytorch/xla/releases
[2] https://pytorch.org/xla/release/2.0/index.html
[3] https://stackoverflow.com/questions/79314991/why-am-i-getting-no-matching-distribution-found-for-torch-xla-2-5-0-nest
[4] https://awsdocs-neuron.readthedocs-hosted.com/en/latest/general/appnotes/torch-neuronx/introducing-pytorch-2-x.html
[5] https://github.com/pytorch/xla
[6] https://github.com/pytorch/xla/issues/4
[7] https://discuss.pytorch.org/t/pytorch-does-not-support-python-3-11/173747
[8] https://pytorch.org/xla/master/gpu.html