nvidia-docker container for https://github.com/1adrianb/human-pose-estimation

Dockerfile

FROM nvidia/cuda:8.0-cudnn5-devel

WORKDIR /root

RUN apt-get update
RUN apt-get upgrade -y
RUN apt-get install -y --no-install-recommends openssl ca-certificates \
	git-core cmake curl wget unzip gfortran libreadline-dev ncurses-dev \
	libgoogle-glog-dev libboost-dev libboost-thread-dev libopenblas-dev

# Core Torch packages
RUN git clone https://github.com/torch/luajit-rocks.git
RUN cd luajit-rocks && mkdir build && cd build \
    && cmake -DCMAKE_INSTALL_PREFIX=/usr -DCMAKE_BUILD_TYPE=Release \
             -DWITH_LUAJIT21=ON -DLUAJIT_ENABLE_LUA52COMPAT=ON .. \
    && make && make install
RUN luarocks install torch
RUN luarocks install trepl

# Common torch package
RUN luarocks install luafilesystem
RUN luarocks install --server=http://luarocks.org/dev luash
RUN luarocks install nn
RUN luarocks install optnet
RUN luarocks install xlua
RUN luarocks install cudnn

# Extra torch package
RUN apt-get install -y --no-install-recommends libjpeg-dev libpng-dev \
	graphicsmagick-imagemagick-compat
RUN luarocks install image
RUN apt-get install -y --no-install-recommends python2.7-dev python-numpy
RUN git clone https://github.com/facebook/thpp.git
RUN cd thpp/thpp && mkdir build && cd build \
    && cmake -DCMAKE_INSTALL_PREFIX=/usr -DCMAKE_BUILD_TYPE=Release \
       -DNO_THRIFT=ON -DNO_FOLLY=ON -DNO_TESTS=ON .. \
    && make && make install
RUN luarocks install fbpython

# Pull project
RUN git clone https://github.com/1adrianb/human-pose-estimation

CMD /bin/bash