install compilers

deepmodeling · Jun 15, 2024 · fe6dd96 · fe6dd96
1 parent 5c3cfe2
commit fe6dd96
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml
@@ -19,7 +19,7 @@ jobs:
     runs-on: nvidia
     # https://github.com/deepmodeling/deepmd-kit/pull/2884#issuecomment-1744216845
     container:
-      image: nvidia/cuda:12.3.1-devel-ubuntu22.04
+      image: ubuntu:24.04
       options: --gpus all
     if: github.repository_owner == 'deepmodeling' && (github.event_name == 'pull_request' && github.event.label && github.event.label.name == 'Test CUDA' || github.event_name == 'workflow_dispatch' || github.event_name == 'merge_group')
     steps:

diff --git a/source/install/cuda_env.yml b/source/install/cuda_env.yml
@@ -12,3 +12,6 @@ dependencies:
   - ninja
   - wget
   - unzip
+  - gcc=12.*
+  - gxx=12.*
+  - cuda-nvcc=12.*