Add Dockerfile and K8S deployments for a basic TPU pytorch workflow.

PiperOrigin-RevId: 620968670
2026-05-22 17:12:49 -07:00 · 2024-04-01 15:25:12 -07:00
parent 8c9cbf0d90
commit 7e9c01f12a
1 changed files with 3 additions and 0 deletions
@@ -0,0 +1,3 @@
+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