opt static linking, Docker CUDA

Author: tmbdev

Dockerfile

@@ -0,0 +1,50 @@
+FROM ubuntu:15.04
+MAINTAINER Tom <[email protected]>
+ENV DEBIAN_FRONTEND noninteractive
+
+RUN echo hi
+RUN apt-get -qqy update
+RUN apt-get -qqy install mercurial
+RUN apt-get -qqy install build-essential 
+RUN apt-get -qqy install g++ gdb swig2.0 scons
+RUN apt-get -qqy install git
+RUN apt-get -qqy install wget
+RUN cd /tmp && wget -nd http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1504/x86_64/cuda-repo-ubuntu1504_7.5-18_amd64.deb
+RUN cd /tmp && dpkg -i cuda-repo-ubuntu1504*.deb
+RUN apt-get -qqy update
+RUN apt-get -qqy install cuda-7-5
+RUN apt-get -qqy install cuda-command-line-tools-7-5
+RUN apt-get -qqy install cuda-core-7-5
+RUN apt-get -qqy install cuda-cublas-7-5
+RUN apt-get -qqy install cuda-cublas-dev-7-5
+RUN apt-get -qqy install cuda-cudart-7-5
+RUN apt-get -qqy install cuda-cudart-dev-7-5
+RUN apt-get -qqy install cuda-cufft-7-5
+RUN apt-get -qqy install cuda-cufft-dev-7-5
+RUN apt-get -qqy install cuda-curand-7-5
+RUN apt-get -qqy install cuda-curand-dev-7-5
+RUN apt-get -qqy install cuda-cusolver-7-5
+RUN apt-get -qqy install cuda-cusolver-dev-7-5
+RUN apt-get -qqy install cuda-cusparse-7-5
+RUN apt-get -qqy install cuda-cusparse-dev-7-5
+RUN apt-get -qqy install cuda-minimal-build-7-5
+RUN apt-get -qqy install cuda-misc-headers-7-5
+RUN apt-get -qqy install cuda-npp-7-5
+RUN apt-get -qqy install cuda-npp-dev-7-5
+RUN apt-get -qqy install cuda-nvrtc-7-5
+RUN apt-get -qqy install cuda-nvrtc-dev-7-5
+RUN apt-get -qqy install cuda-runtime-7-5
+RUN apt-get -qqy install cuda-toolkit-7-5
+RUN apt-get -qqy install cuda-visual-tools-7-5
+RUN apt-get -qqy install cuda-samples-7-5
+
+RUN apt-get -qqy install apt-utils
+RUN apt-get -qqy install protobuf-compiler libprotobuf-dev
+RUN apt-get -qqy install libpng12-dev
+RUN cd /usr/local/include && hg clone http://bitbucket.org/eigen/eigen eigen3
+RUN apt-get -qqy install strace
+
+RUN apt-get clean && rm -rf /tmp/* /var/lib/apt/lists/* /var/tmp/*
+
+VOLUME /work
+WORKDIR /work

Makefile

@@ -0,0 +1,3 @@
+DONE: Dockerfile
+	docker build -t tmbdev/ubuntu-cuda .
+	touch DONE

SConstruct

@@ -53,7 +53,7 @@ if option("threads", 1):
   env.Append(CPPDEFINES={'EIGEN_USE_THREADS': '1'})
   env.Append(LIBS=["pthread"])
 
-if option("gpu", 1):
+if option("gpu", 0):
   env.Append(CPPDEFINES={'EIGEN_USE_GPU': '1'})
   env.Append(LIBS=["cublas","cuda","cudart"])
 
@@ -95,6 +95,18 @@ else:
     env.Append(CXXFLAGS="-g -O3".split())
     env.Append(CCFLAGS="-g".split())
 
+# Hack for linking some libraries statically.
+
+abslibs = """/usr/lib/x86_64-linux-gnu/libpng12.a
+/usr/lib/x86_64-linux-gnu/libprotobuf.a
+/usr/lib/x86_64-linux-gnu/libz.a"""
+
+if not option("static", 0):
+    env.Append(LIBS=["png", "protobuf"])
+else:
+    abslibs = [File(x) for x in abslibs.split()]
+    env.Append(LIBS=abslibs)
+
 # Extra layers (old layers or testing)
 
 if option("extras", 0):
@@ -121,8 +133,6 @@ if option("display", 0):
 else:
     env.Append(CPPDEFINES={'NODISPLAY': 1})
 
-env.Append(LIBS=["png", "protobuf"])
-
 # We need to compile the protocol buffer definition as part of the build.
 
 env.Protoc("clstm.proto")

curun

@@ -0,0 +1,6 @@
+#!/bin/sh
+for dev in /dev/nv*; do
+  devs="$devs --device=$dev:$dev"
+done
+uid=${uid:-$(id -u)}
+docker run --privileged --rm -u $uid -v $(pwd):/work -w=/work -e PATH=.:/usr/local/cuda/bin:/usr/local/bin:/usr/bin:/bin -i -t tmbdev/ubuntu-cuda "$@"