Internal change.

PiperOrigin-RevId: 689046241
google · Oct 23, 2024 · c3ffeb1 · c3ffeb1
1 parent 74f6136
commit c3ffeb1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/images/tpu/pytorch/Dockerfile.x86_64 b/images/tpu/pytorch/Dockerfile.x86_64
@@ -1,3 +1,5 @@
 FROM python:3.10
 
-RUN pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 torchvision -f https://storage.googleapis.com/libtpu-releases/index.html
+RUN pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 torchvision -f https://storage.googleapis.com/libtpu-releases/index.html
+
+RUN git clone --depth=1 --branch r2.0 https://github.com/pytorch/xla.git