diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 9a9b6a5..ae9894b 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -16,6 +16,8 @@ jobs:
   - run: flake8 run.py core
  test:
   runs-on: ubuntu-latest
+  container:
+   image: nvidia/cuda:11.8.0-devel-ubuntu22.04
   strategy:
    matrix:
     include: