From b91bcde3e7ff06f288353fe26d0352704b780ccd Mon Sep 17 00:00:00 2001 From: Byungheong Jeong <byungheon.jeong@gmail.com> Date: Thu, 4 Jun 2020 01:47:00 -0700 Subject: [PATCH] Update Dockerfile --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index a556b9a..b6e311f 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 -- GitLab