Skip to content

Commit d721f7c

Browse files
authored
CI: add GPU resource flag to container service
1 parent 45e8781 commit d721f7c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

.github/workflows/cuda.yml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,9 @@ jobs:
3232
name: Do the job on the runner
3333
needs: start-runner # required to start the main job when the runner is ready
3434
runs-on: ${{ needs.start-runner.outputs.label }} # run the job on the newly created runner
35-
container: ghcr.io/deepmodeling/abacus-development-kit:cuda
35+
container:
36+
image: ghcr.io/deepmodeling/abacus-development-kit:cuda
37+
options: --gpus all
3638
steps:
3739
- name: Checkout
3840
uses: actions/checkout@v3

0 commit comments

Comments
 (0)