debug - with a container?

huggingface · Feb 27, 2024 · 5b55da5 · 5b55da5
1 parent 9b7ca11
commit 5b55da5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/test_tpu_tgi.yml b/.github/workflows/test_tpu_tgi.yml
@@ -19,6 +19,9 @@ jobs:
   do-the-job:
     name: Run TGI tests
     runs-on: [optimum-tpu] # run the configured Google TPU
+    container:
+      image: ubuntu:22.04
+      options: --shm-size "16gb" --net host --privileged
     env:
       PJRT_DEVICE: TPU
     steps:
@@ -27,6 +30,7 @@ jobs:
       - name: Create venv and install Pytorch/XLA
         run: |
           echo "PJRT_DEVICE set to $PJRT_DEVICE"
+          sudo apt install python3.10-venv -y
           echo "Creating virtual environment..."
           python3 -m venv venv-tpu-pytorch
           source venv-tpu-pytorch/bin/activate