From 93aed417d9bc1ca9d38ae3cb8ff6830befec050c Mon Sep 17 00:00:00 2001 From: akshay bhat Date: Thu, 22 Feb 2018 23:48:24 -0800 Subject: [PATCH] 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 | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deploy/dockerfiles/Dockerfile.gpu b/deploy/dockerfiles/Dockerfile.gpu index c0081d85..4fb487f2 100644 --- a/deploy/dockerfiles/Dockerfile.gpu +++ b/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 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