gvisor.dev/gvisor@v0.0.0-20240520182842-f9d4d51c7e0f/images/tpu/pytorch/Dockerfile.x86_64 (about)

     1  FROM python:3.10
     2  
     3  RUN pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 torchvision -f https://storage.googleapis.com/libtpu-releases/index.html