Backport PR #2437: Switch cuda docker image tag (#2439)

Co-authored-by: Martin Kim <[email protected]>
scverse · Jan 25, 2024 · 24aeaaa · 24aeaaa
1 parent 4ae86e3
commit 24aeaaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test_linux_cuda.yml b/.github/workflows/test_linux_cuda.yml
@@ -26,7 +26,7 @@ jobs:
         cuda: ["11"]
 
     container:
-      image: martinkim0/scvi-tools:py${{ matrix.python }}-cu${{ matrix.cuda }}-base
+      image: scverse/scvi-tools:py${{ matrix.python }}-cu${{ matrix.cuda }}-base
       options: --user root --gpus all
 
     environment: