Fix base image

YosefLab · Sep 11, 2023 · 6ee262f · 6ee262f
1 parent 3e307fe
commit 6ee262f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/images/linux/cuda/autotune/Dockerfile b/images/linux/cuda/autotune/Dockerfile
@@ -1,4 +1,4 @@
-FROM martinkim0/scvi-tools:py${PYTHON_VERSION}-cu${CUDA_VERSION}-base
+FROM martinkim0/scvi-tools:py3.11-cu11-base
 
 RUN pip install "scvi-tools[autotune,census] @ git+https://github.com/scverse/scvi-tools@main" && \
     pip cache purge