human-pose-estimation

human-pose-estimation copied to clipboard

I have found it rather tedious to deploy torch and its dependencies on my computer. Here is a dockerfile I wrote to do the job pretty much automatically (note that it requires the nvidia wrapper to docker):

https://gist.github.com/pixelou/e06d8b049001254575d745be2b2b0402

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

deploy like so:

curl -O https://gist.githubusercontent.com/pixelou/e06d8b049001254575d745be2b2b0402/raw/9585f22ba540a2fb07b54c4c3b8e314a53010199/Dockerfile
nvidia-docker build -t humposest -f ./Dockerfile
nvidia-docker run -it humposest bash

@pixelou thanks a lot for your contribution, it will definitely come in handy. If you want to make a pull request I am happy to add it to the instructions.

Hello, Sure, I will use it for a few days to see wether I forgot any dependencies and then prepare a PR.