diff --git a/.github/workflows/run-linux-cuda.yml b/.github/workflows/run-linux-cuda.yml
index e86dea44..29ffe328 100644
--- a/.github/workflows/run-linux-cuda.yml
+++ b/.github/workflows/run-linux-cuda.yml
@@ -13,7 +13,7 @@ jobs:
     runs-on: [self-hosted, Linux, X64, CUDA]
 
     container:
-      image: martinkim0/scvi-tools:ubuntu-latest-mamba-latest-python-3.11-cuda-11
+      image: martinkim0/scvi-tools:py3.11-cu11
       options: --user root --gpus all
 
     steps:
@@ -22,11 +22,15 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install -U scvi-tools[tutorials]
+          python -m pip install -U scvi-tools[tutorials]
 
       - name: Install nbconvert
         run: |
-          pip install -U nbconvert
+          python -m pip install -U jupyter ipykernel nbconvert
+
+      - name: Install kernel
+        run: |
+          python -m ipykernel install
 
       - name: Run notebook
         run: |