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 @@ -50,15 +50,24 @@ two C++ ABI flavors: C++11 and pre-C++11. Pre-C++11 is the default to align with
50
50
PyTorch upstream, but C++11 ABI wheels and docker images have better lazy tensor
51
51
tracing performance.
52
52
53
- To install C++11 ABI flavored 2.6 wheels:
53
+ To install C++11 ABI flavored 2.6 wheels (Python 3.10 example) :
54
54
55
55
``` sh
56
- pip install torch==2.6.0+cpu.cxx11.abi torch_xla[tpu]==2.6.0+cxx11 \
56
+ pip install torch==2.6.0+cpu.cxx11.abi \
57
+ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0%2Bcxx11-cp310-cp310-manylinux_2_28_x86_64.whl \
58
+ ' torch_xla[tpu]' \
57
59
-f https://storage.googleapis.com/libtpu-releases/index.html \
58
60
-f https://storage.googleapis.com/libtpu-wheels/index.html \
59
61
-f https://download.pytorch.org/whl/torch
60
62
```
61
63
64
+ The above command works for Python 3.10. We additionally have Python 3.9 and 3.11
65
+ wheels:
66
+
67
+ - 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
68
+ - 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
69
+ - 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
70
+
62
71
To access C++11 ABI flavored docker image:
63
72
64
73
```
You can’t perform that action at this time.
0 commit comments