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