From 5a341844cb0d8d5935bea4d9cec252feee4b7310 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 3 Oct 2023 02:52:16 -0400 Subject: [PATCH] run Test CUDA in container See https://github.com/deepmodeling/deepmd-kit/pull/2884#issuecomment-1744216845 Signed-off-by: Jinzhe Zeng --- .github/workflows/test_cuda.yml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml index adc20c27a9..2850290d85 100644 --- 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