From ae566a2461b6cd58e7bed9f008fc7ad28263b4d1 Mon Sep 17 00:00:00 2001 From: Christopher Heiny Date: Tue, 6 Jun 2017 15:08:17 -0700 Subject: [PATCH 1/4] Fix OpenCV build - add -DCMAKE_LIBRARY_PATH=/usr/local/cuda/lib64/stubs to the cmake command line. --- Dockerfile.gpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 3bad5a4..3957e2f 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -224,7 +224,7 @@ RUN git clone --depth 1 https://github.com/opencv/opencv.git /root/opencv && \ cd /root/opencv && \ mkdir build && \ cd build && \ - cmake -DWITH_QT=ON -DWITH_OPENGL=ON -DFORCE_VTK=ON -DWITH_TBB=ON -DWITH_GDAL=ON -DWITH_XINE=ON -DBUILD_EXAMPLES=ON .. && \ + cmake -DWITH_QT=ON -DWITH_OPENGL=ON -DCMAKE_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/ -DFORCE_VTK=ON -DWITH_TBB=ON -DWITH_GDAL=ON -DWITH_XINE=ON -DBUILD_EXAMPLES=ON .. && \ make -j"$(nproc)" && \ make install && \ ldconfig && \ From 9cb1c98ab05570521a83c9ff7978db2506bfafb4 Mon Sep 17 00:00:00 2001 From: Elias Obeid Date: Thu, 30 Mar 2017 16:11:38 +0400 Subject: [PATCH 2/4] install specific html5lib so TensorBoard can run --- Dockerfile.gpu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 3957e2f..ae72003 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -136,6 +136,8 @@ RUN pip --no-cache-dir install --upgrade ipython && \ RUN pip --no-cache-dir install \ https://storage.googleapis.com/tensorflow/linux/${TENSORFLOW_ARCH}/tensorflow_${TENSORFLOW_ARCH}-${TENSORFLOW_VERSION}-cp27-none-linux_x86_64.whl +# Install specific html5lib so TensorBoard can run +RUN pip --no-cache-dir install --upgrade html5lib==1.0b8 # Install dependencies for Caffe RUN apt-get update && apt-get install -y \ From 8dcfd6cc1ef726d0918261370eba505b6e7514bb Mon Sep 17 00:00:00 2001 From: Christopher Heiny Date: Tue, 6 Jun 2017 17:53:34 -0700 Subject: [PATCH 3/4] Added tensorflow-gpu package. --- Dockerfile.gpu | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.gpu b/Dockerfile.gpu index ae72003..13d9ad3 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -135,6 +135,7 @@ RUN pip --no-cache-dir install --upgrade ipython && \ # Install TensorFlow RUN pip --no-cache-dir install \ https://storage.googleapis.com/tensorflow/linux/${TENSORFLOW_ARCH}/tensorflow_${TENSORFLOW_ARCH}-${TENSORFLOW_VERSION}-cp27-none-linux_x86_64.whl +RUN pip install --upgrade tensorflow-gpu # Install specific html5lib so TensorBoard can run RUN pip --no-cache-dir install --upgrade html5lib==1.0b8 From 32caf7b65df97854765107f01ade0aae4b80ea80 Mon Sep 17 00:00:00 2001 From: Christopher Heiny Date: Thu, 15 Jun 2017 14:28:35 -0700 Subject: [PATCH 4/4] Update to latest Keras version. Use luarocks torch install to fix build problems. --- Dockerfile.gpu | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 13d9ad3..2e72778 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -5,7 +5,7 @@ MAINTAINER Sai Soundararaj ARG THEANO_VERSION=rel-0.8.2 ARG TENSORFLOW_VERSION=0.12.1 ARG TENSORFLOW_ARCH=gpu -ARG KERAS_VERSION=1.2.0 +ARG KERAS_VERSION=2.0.5 ARG LASAGNE_VERSION=v0.1 ARG TORCH_VERSION=latest ARG CAFFE_VERSION=master @@ -140,6 +140,9 @@ RUN pip install --upgrade tensorflow-gpu # Install specific html5lib so TensorBoard can run RUN pip --no-cache-dir install --upgrade html5lib==1.0b8 +# Include Keras neural net library for use with Theano and TensorFlow. +RUN pip install --upgrade keras # force rebuild + # Install dependencies for Caffe RUN apt-get update && apt-get install -y \ libboost-all-dev \ @@ -210,6 +213,7 @@ ENV LUA_CPATH='/root/torch/install/lib/?.so;'$LUA_CPATH # Install the latest versions of nn, cutorch, cunn, cuDNN bindings and iTorch RUN luarocks install nn && \ + luarocks install torch && \ luarocks install cutorch && \ luarocks install cunn && \ luarocks install loadcaffe && \