ADD Docker (#305)

* add docker

Dockerfile

@@ -0,0 +1,21 @@
+FROM nvcr.io/nvidia/pytorch:19.05-py3
+
+# Set working directory	# Set working directory
+WORKDIR /workspace
+
+#ENV PYTHONPATH "${PYTHONPATH}:/workspace"
+
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y python3-tk python-pip git tmux htop tree
+
+RUN python -m pip install --upgrade pip
+RUN python -m pip install torch==1.4.0
+RUN python -m pip install torchvision==0.5.0
+RUN python -m pip install pycocotools==2.0.0
+
+#RUN chmod a+rwx -R /opt/conda/
+
+COPY ./setup.py .
+COPY ./encoding ./encoding
+RUN python setup.py develop
+
+COPY ./experiments ./experiments

scripts/build_docker.sh

@@ -0,0 +1 @@
+docker build -t encoding .

scripts/run_docker.sh

@@ -0,0 +1 @@
+docker run --rm -it --ulimit memlock=-1 --ulimit stack=67108864 -v $COCO_DIR:/coco --ipc=host --gpus=all encoding