Public Repository

Last pushed: a year ago
Short Description
Torch + Terra + MPI installation
Full Description

FROM nimbix/base-ubuntu-nvidia-openmpi:8.0-cudnn5-devel-ubuntu16.04

RUN apt-get update && \
apt-get -y install numactl subversion groff && \
apt-get clean

WORKDIR /tmp
RUN git clone https://github.com/torch/luajit-rocks.git && mkdir -p luajit-rocks/build

WORKDIR /tmp/luajit-rocks/build
RUN export LUAROCKS_INSTALL_DIR=/usr/local/; \
cmake .. -DCMAKE_INSTALL_PREFIX=${LUAROCKS_INSTALL_DIR} -DCMAKE_BUILD_TYPE=Release -DWITH_LUAJIT21=ON && \
make install; \
export PATH=${PATH}:/usr/local/bin; \
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib64; \
luarocks install luaffi && \
luarocks install torchnet && \
luarocks install mnist

cutorch takes forever to build, just put it here by itself

RUN export PATH=${PATH}:/usr/local/bin; \
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib64; \
luarocks install cutorch

RUN export PATH=${PATH}:/usr/local/bin; \
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib64; \
luarocks install cunn

RUN cd /tmp && \
svn co http://llvm.org/svn/llvm-project/llvm/tags/RELEASE_352/final/ llvm && \
cd /tmp/llvm/tools && \
svn co http://llvm.org/svn/llvm-project/cfe/tags/RELEASE_352/final/ clang && \
cd /tmp/llvm/projects && \
svn co http://llvm.org/svn/llvm-project/compiler-rt/tags/RELEASE_352/final/ compiler-rt && \
mkdir -p /tmp/llvm_build && \
cd /tmp/llvm_build && \
../llvm/configure --prefix=/usr/local/llvm --enable-optimized=YES --enable-assertions=YES --enable-cxx1y && \
make -j 16 && \
make install

RUN LLVM_CONFIG=/usr/local/llvm/bin/llvm-config CLANG=/usr/local/llvm/bin/clang luarocks install terra

RUN apt-get update && apt-get install -y numactl bc

RUN ln -s /usr /usr/local/mpi
RUN ln -s /usr/local/libexec/osu-micro-benchmarks /usr/local/osu-micro-benchmarks
RUN chmod -R 777 /usr/local

Docker Pull Command
Owner
nicolasvasilache