Let's try another image then

Signed-off-by: Huy Do <huydhn@gmail.com>

Author: huydhn

.github/workflows/flash_attention.yml

@@ -18,10 +18,9 @@ jobs:
     runs-on: linux.dgx.b200.8
     container:
       # https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/
-      image: nvcr.io/nvidia/pytorch:25.06-py3
+      # image: nvcr.io/nvidia/pytorch:25.06-py3
+      image: pytorch/pytorch:2.7.1-cuda12.8-cudnn9-runtime
       options: --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864
-      volumes:
-        - /home/runner/externals:/__e
     steps:
       - uses: actions/checkout@v4
         with: