File tree 1 file changed +11
-2
lines changed
1 file changed +11
-2
lines changed Original file line number Diff line number Diff line change @@ -57,15 +57,24 @@ two C++ ABI flavors: C++11 and pre-C++11. Pre-C++11 is the default to align with
57
57
PyTorch upstream, but C++11 ABI wheels and docker images have better lazy tensor
58
58
tracing performance.
59
59
60
- To install C++11 ABI flavored 2.6 wheels:
60
+ To install C++11 ABI flavored 2.6 wheels (Python 3.10 example) :
61
61
62
62
``` sh
63
- pip install torch==2.6.0+cpu.cxx11.abi torch_xla[tpu]==2.6.0+cxx11 \
63
+ pip install torch==2.6.0+cpu.cxx11.abi \
64
+ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0%2Bcxx11-cp310-cp310-manylinux_2_28_x86_64.whl \
65
+ ' torch_xla[tpu]' \
64
66
-f https://storage.googleapis.com/libtpu-releases/index.html \
65
67
-f https://storage.googleapis.com/libtpu-wheels/index.html \
66
68
-f https://download.pytorch.org/whl/torch
67
69
```
68
70
71
+ The above command works for Python 3.10. We additionally have Python 3.9 and 3.11
72
+ wheels:
73
+
74
+ - 3.9: https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0%2Bcxx11-cp39-cp39-manylinux_2_28_x86_64.whl
75
+ - 3.10: https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0%2Bcxx11-cp310-cp310-manylinux_2_28_x86_64.whl
76
+ - 3.11: https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0%2Bcxx11-cp311-cp311-manylinux_2_28_x86_64.whl
77
+
69
78
To access C++11 ABI flavored docker image:
70
79
71
80
```
You can’t perform that action at this time.
0 commit comments