Update Dockerfile

PGerardi · web-flow · commit b7cf15b5abf3 · 2023-09-07T14:54:45.000+02:00
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

Original file line number	Diff line number	Diff line change
`@@ -17,3 +17,4 @@ ENV PATH=/opt/conda/bin:$PATH`
`17`	`17`
`18`	`18`	`# Install CUDA.`
`19`	`19`	`RUN conda install --channel nvidia --yes cuda-runtime="$CUDA_VERSION"`
	`20`	`+RUN conda install -c conda-forge cudnn`