Add compiler to let users install pip packages that need compilation

pydatasci-gpu/Dockerfile

@@ -11,7 +11,7 @@ RUN wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubu
 RUN apt-get update && \
     apt-get -y install cuda9.0 cuda-cublas-9-0 cuda-cufft-9-0 cuda-curand-9-0 \
       cuda-cusolver-9-0 cuda-cusparse-9-0 libcudnn7=7.1.4.18-1+cuda9.0 \
-      libnccl2=2.2.13-1+cuda9.0 cuda-command-line-tools-9-0 && \
+      libnccl2=2.2.13-1+cuda9.0 cuda-command-line-tools-9-0 build-essential && \
     apt-get clean
 
 RUN pip install --upgrade pip \

pydatasci/Dockerfile

@@ -23,6 +23,8 @@ RUN pip install --upgrade pip \
             torch==0.4.1 \
             torchvision
 
+RUN apt-get -y install build-essential && apt-get clean
+
 COPY start_notebook.sh /usr/local/bin/start_notebook.sh
 RUN chmod 755 /usr/local/bin/start_notebook.sh