Add dockerfile to build docker image permanently

Dockerfile

@@ -0,0 +1,11 @@
+FROM nvcr.io/nvidia/pytorch:20.11-py3
+
+LABEL maintainer="Haneol Kim <bestwook7@gmail.com>"
+
+RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
+RUN pip install seaborn thop
+
+RUN cd /
+RUN git clone https://github.com/JunnYu/mish-cuda && cd mish-cuda && python setup.py build install
+
+RUN cd / && git clone https://github.com/fbcotter/pytorch_wavelets && cd pytorch_wavelets && pip install .

README.md

@@ -59,33 +59,11 @@ Docker environment (recommended)
 <details><summary> <b>Expand</b> </summary>
 
 ```
-# create the docker container, you can change the share memory size if you have more.
-nvidia-docker run --name yolor -it -v your_coco_path/:/coco/ -v your_code_path/:/yolor --shm-size=64g nvcr.io/nvidia/pytorch:20.11-py3
+# Build docker
+docker build -t yolor:latest .
 
-# apt install required packages
-apt update
-apt install -y zip htop screen libgl1-mesa-glx
-
-# pip install required packages
-pip install seaborn thop
-
-# install mish-cuda if you want to use mish activation
-# https://github.com/thomasbrandon/mish-cuda
-# https://github.com/JunnYu/mish-cuda
-cd /
-git clone https://github.com/JunnYu/mish-cuda
-cd mish-cuda
-python setup.py build install
-
-# install pytorch_wavelets if you want to use dwt down-sampling module
-# https://github.com/fbcotter/pytorch_wavelets
-cd /
-git clone https://github.com/fbcotter/pytorch_wavelets
-cd pytorch_wavelets
-pip install .
-
-# go to code folder
-cd /yolor
+# Run docker
+docker run -tid --privileged --gpus all -e DISPLAY=${DISPLAY} -e TERM=xterm-256color -v /tmp/.X11-unix/:/tmp/.X11-unix:ro -v your_coco_path/:/coco/ -v your_code_path/:/yolor --shm-size=64g yolor:latest /bin/bash
 ```
 
 </details>