We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e90dd10 commit 9126b85Copy full SHA for 9126b85
1 file changed
tpu/config.txt
@@ -1,7 +1,7 @@
1
-BASE_IMAGE=python:3.11
2
-PYTHON_WHEEL_VERSION=cp311
3
-PYTHON_VERSION_PATH=python3.11
4
-TENSORFLOW_VERSION=2.19.0
+BASE_IMAGE=python:3.10
+PYTHON_WHEEL_VERSION=cp310
+PYTHON_VERSION_PATH=python3.10
+TENSORFLOW_VERSION=2.18
5
# gsutil ls gs://pytorch-xla-releases/wheels/tpuvm/* | grep libtpu | grep torch_xla | grep -v -E ".*rc[0-9].*" | sed 's/.*torch_xla-\(.*\)+libtpu.*/\1/' | sort -rV
6
# Supports nightly
7
TORCH_VERSION=2.6.0
0 commit comments