Fixes 1adrianb#13

Dockerfile

@@ -1,5 +1,5 @@
 # Based on https://github.com/pytorch/pytorch/blob/master/Dockerfile
-FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 
+FROM nvidia/cuda:8.0-cudnn5-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
 
@@ -10,6 +10,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
          curl \
          vim \
          ca-certificates \
+         libboost-all-dev \
          libjpeg-dev \
          libpng-dev &&\
      rm -rf /var/lib/apt/lists/*
@@ -31,4 +32,4 @@ RUN chmod -R a+w /workspace
 RUN git clone https://github.com/1adrianb/face-alignment
 WORKDIR /workspace/face-alignment
 RUN pip install -r requirements.txt
-RUN python setup.py install
+RUN python setup.py install