Turning off rerun and check GPU information

probcomp · Jul 2, 2024 · 4a6ff5f · 4a6ff5f
1 parent 483213d
commit 4a6ff5f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -15,6 +15,7 @@ env:
   XLA_FLAGS: --xla_gpu_enable_command_buffer=
   XLA_PYTHON_CLIENT_PREALLOCATE: "false"
   XLA_PYTHON_CLIENT_ALLOCATOR: "platform"
+  RERUN: "off"
 
 jobs:
   test:
@@ -69,6 +70,11 @@ jobs:
       - name: Show installed packages (for debugging)
         run: pip list
 
+      - name: Show NVIDIA GPU information (for debugging)
+        run: |
+          nvidia-smi
+          nvcc --version
+
       - name: Trigger compilation of NVDiffrast
         run: python -c "from b3d.renderer.nvdiffrast.jax import _get_plugin; _get_plugin(gl=True)"