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