Update Dockerfile

superlinear-ai · Sep 7, 2023 · b7cf15b · b7cf15b
1 parent ec4eac7
commit b7cf15b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -17,3 +17,4 @@ ENV PATH=/opt/conda/bin:$PATH
 
 # Install CUDA.
 RUN conda install --channel nvidia --yes cuda-runtime="$CUDA_VERSION"
+RUN conda install -c conda-forge cudnn