Upgrade GPU docker container.

Eventually:

- Export models as a zip file.

- Implement training detection model using YOLO

- Implement location net and clustering using PQKMeans

- Implement fine-tuning / training face recognition models

- Pre-condition checks, and failed to launch error messages for processes.

deploy/dockerfiles/Dockerfile.gpu

@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.4.0-gpu
+FROM tensorflow/tensorflow:1.6.0-rc1-devel-gpu
 MAINTAINER Akshay Bhat <[email protected]>
 RUN apt-get update -q && apt-get install -y wget xz-utils
 WORKDIR "/bin/"
@@ -22,8 +22,8 @@ RUN pip install --upgrade cffi
 RUN pip install --no-deps keras
 RUN pip install -r requirements.txt
 RUN pip install pip --upgrade
-RUN pip install --no-cache-dir http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl
-RUN pip install --no-cache-dir mxnet-cu80==0.11.0
+RUN pip install --no-cache-dir http://download.pytorch.org/whl/cu90/torch-0.3.1-cp27-cp27mu-linux_x86_64.whl
+RUN pip install --no-cache-dir mxnet-cu90
 RUN pip install torchvision
 RUN apt-get install -y libhdf5-dev
 RUN pip install --no-deps h5py