run Test CUDA in container

See #2884 (comment) Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Oct 3, 2023 · 5a34184 · 5a34184
1 parent f256dff
commit 5a34184
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml
@@ -9,6 +9,10 @@ jobs:
   test_cuda:
     name: Test Python and C++ on CUDA
     runs-on: nvidia
+    # https://github.com/deepmodeling/deepmd-kit/pull/2884#issuecomment-1744216845
+    container:
+      image: python:3.11
+      options: --gpus all
     if: github.repository_owner == 'deepmodeling' && github.event.label.name == 'Test CUDA' || github.event_name == 'workflow_dispatch'
     steps:
     - uses: actions/checkout@v4