add a dockerfile

.dockerignore

+*.zip
+.git
+datasets/linemod/Linemod_preprocessed
+datasets/ycb/YCB_Video_Dataset

Dockerfile

+FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+# Essentials: developer tools, build tools, OpenBLAS
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    apt-utils git curl vim unzip openssh-client wget \
+    build-essential cmake \
+    libopenblas-dev
+
+# Python 3.5
+# For convenience, alias (but don't sym-link) python & pip to python3 & pip3 as recommended in:
+# http://askubuntu.com/questions/351318/changing-symlink-python-to-python3-causes-problems
+RUN apt-get update && apt-get install -y --no-install-recommends python3.5 python3.5-dev python3-pip python3-tk && \
+    pip3 install --no-cache-dir --upgrade pip setuptools && \
+    echo "alias python='python3'" >> /root/.bash_aliases && \
+    echo "alias pip='pip3'" >> /root/.bash_aliases
+# Pillow and it's dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends libjpeg-dev zlib1g-dev && \
+    pip3 --no-cache-dir install Pillow
+# Science libraries and other common packages
+RUN pip3 --no-cache-dir install \
+    numpy scipy pyyaml cffi pyyaml matplotlib Cython requests
+
+# Tensorflow 1.11 - GPU
+RUN pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
+RUN pip3 install torchvision==0.2.2.post3
+
+# Expose port for TensorBoard
+EXPOSE 6006
+
+# cd to home on login
+RUN echo "cd /root/dense_fusion" >> /root/.bashrc 

build.sh

+#!/bin/bash
+
+#docker build --no-cache -t hri -f Dockerfile .
+docker build -t dense_fusion -f Dockerfile .

run.sh

+#!/bin/sh
+
+# Runs a docker container with the image created by build.sh
+until nvidia-docker ps
+do
+    echo "Waiting for docker server"
+    sleep 1
+done
+
+XSOCK=/tmp/.X11-unix
+
+XAUTH=/root/.Xauthority
+
+SRC_CONTAINER=/root/dense_fusion
+SRC_HOST="$(pwd)"
+
+xhost local:root
+
+nvidia-docker run \
+    --name dense_fusion \
+    -it --rm \
+    --volume=$XSOCK:$XSOCK:rw \
+    --volume=$XAUTH:$XAUTH:rw \
+    --volume=$SRC_HOST:$SRC_CONTAINER:rw \
+    --env="XAUTHORITY=${XAUTH}" \
+    --env="DISPLAY=${DISPLAY}" \
+    --privileged -v /dev/bus/usb:/dev/bus/usb \
+    --net=host \
+    dense_fusion