Create a TPU hello world image for use in gVisor tests.

PiperOrigin-RevId: 574506997
This commit is contained in:
Lucas Manning
2023-10-18 09:57:07 -07:00
committed by gVisor bot
parent d6e83e2802
commit fb487fbb2f
+4
View File
@@ -0,0 +1,4 @@
FROM python:3.8
RUN pip install --upgrade jax
RUN pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html