install vision in devel dockerfile, minor fixes to dockerfile (pytorch#2090)

Author: ngimel

Dockerfile

@@ -1,18 +1,16 @@
-FROM nvidia/cuda:8.0-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
 
-ENV CUDNN_VERSION 6.0.20 
 RUN apt-get update && apt-get install -y --no-install-recommends \
          build-essential \
          cmake \
          git \
          curl \
+         vim \
          ca-certificates \
          libjpeg-dev \
-         libpng-dev \
-         libcudnn6=$CUDNN_VERSION-1+cuda8.0 \             
-         libcudnn6-dev=$CUDNN_VERSION-1+cuda8.0 && \
+         libpng-dev &&\
      rm -rf /var/lib/apt/lists/*
 
 RUN curl -o ~/miniconda.sh -O  https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh  && \
@@ -32,5 +30,7 @@ RUN TORCH_CUDA_ARCH_LIST="3.5 5.2 6.0 6.1+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compr
     CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" \
     pip install -v .
 
+RUN git clone https://github.com/pytorch/vision.git && cd vision && pip install -v .
+
 WORKDIR /workspace
 RUN chmod -R a+w /workspace