diff --git a/Dockerfile b/Dockerfile index a556b9afa5b9794fc0758beece7505eeaeefb236..b6e311f5bc637589c647a45a19942978e9027c2e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,6 +2,8 @@ ARG cuda_version=10.0 ARG cudnn_version=7 FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel +WORKDIR /userdata/kerasData + # Install system packages RUN apt-get update && apt-get install -y --no-install-recommends \ bzip2 \ @@ -80,4 +82,3 @@ ENV LANG=C.UTF-8 ENV PYTHONPATH='/src/:$PYTHONPATH' -WORKDIR /userdata/kerasData