Fixes dockerfile

resnet152/dockerfile

@@ -1,6 +1,6 @@
 
-FROM nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04
-    
+FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
+
 RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
 
 USER root
@@ -13,8 +13,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         build-essential \
         ca-certificates \
         cmake \
-        wget \
         curl \
+        wget \
         git && \
         rm -rf /var/lib/apt/lists/*
 
@@ -24,17 +24,16 @@ RUN curl -o ~/miniconda.sh -O  https://repo.continuum.io/miniconda/Miniconda3-la
     ~/miniconda.sh -b -p /opt/conda && \
     rm ~/miniconda.sh && \
     /opt/conda/bin/conda create -y --name py$PYTHON_VERSION python=$PYTHON_VERSION numpy pyyaml scipy \
-    ipython mkl pandas jupyter ipykernel scikit-learn && \
+    ipython pandas scikit-learn && \
     /opt/conda/bin/conda clean -ya
-    
-ENV PATH /opt/conda/envs/py$PYTHON_VERSION/bin:/opt/conda/bin:$PATH
-ENV LD_LIBRARY_PATH /opt/conda/envs/py3.5/lib:$LD_LIBRARY_PATH
+
+ENV PATH /opt/conda/envs/py$PYTHON_VERSION/bin:$PATH
+ENV LD_LIBRARY_PATH /opt/conda/envs/py$PYTHON_VERSION/lib:/usr/local/cuda/lib64/:$LD_LIBRARY_PATH
 
 RUN pip install --upgrade pip && \
-    pip install tensorflow-gpu==1.6.0rc0 && \       
-    conda clean -yt
+    pip install tensorflow-gpu==1.4.1 && \
+    pip install keras==2.1.5 && \
+    pip install -r /code/requirements.txt && \
+    /opt/conda/bin/conda clean -yt
 
-RUN pip install -r /code/requirements.txt
-
-EXPOSE 8888
 EXPOSE 5000