fix docker

Signed-off-by: zethson <lukas.heumos@posteo.net>
mlf-core · Apr 23, 2021 · 4d66b07 · 4d66b07
1 parent 8863036
commit 4d66b07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...core/create/templates/mlflow/mlflow_tensorflow/{{ cookiecutter.project_slug }}/Dockerfile b/...core/create/templates/mlflow/mlflow_tensorflow/{{ cookiecutter.project_slug }}/Dockerfile
@@ -1,9 +1,5 @@
 FROM mlfcore/base:1.2.0
 
-# Tensorflow has issues finding libcudnn8 from the base NVIDIA Container so we reinstall it
-RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
-RUN apt install ./libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
-
 # Install the conda environment
 COPY environment.yml .
 RUN conda env create -f environment.yml && conda clean -a
@@ -17,3 +13,7 @@ RUN conda env export --name {{ cookiecutter.project_slug }} > {{ cookiecutter.pr
 
 # Currently required, since mlflow writes every file as root!
 USER root
+
+# Tensorflow has issues finding libcudnn8 from the base NVIDIA Container so we reinstall it
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
+RUN apt install ./libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
diff --git a/...tter.project_slug_no_hyphen }}/Dockerfile → ...{ cookiecutter.project_slug }}/Dockerfile b/...tter.project_slug_no_hyphen }}/Dockerfile → ...{ cookiecutter.project_slug }}/Dockerfile