From 7e9c01f12a1498657682fe3f54662848abff2db8 Mon Sep 17 00:00:00 2001
From: Jing Chen <chjing@google.com>
Date: Mon, 1 Apr 2024 15:21:44 -0700
Subject: [PATCH] Add Dockerfile and K8S deployments for a basic TPU pytorch
 workflow.

PiperOrigin-RevId: 620968670
---
 images/tpu/pytorch/Dockerfile.x86_64 | 3 +++
 1 file changed, 3 insertions(+)
 create mode 100644 images/tpu/pytorch/Dockerfile.x86_64

diff --git a/images/tpu/pytorch/Dockerfile.x86_64 b/images/tpu/pytorch/Dockerfile.x86_64
new file mode 100644
index 000000000..7da256d9a
--- /dev/null
+++ b/images/tpu/pytorch/Dockerfile.x86_64
@@ -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
\ No newline at end of file