From 50827c432e8208ed1bbd4ab23aa206dd623f17ee Mon Sep 17 00:00:00 2001 From: penolove Date: Wed, 8 Jul 2020 19:56:10 +0800 Subject: [PATCH] add docker example --- .gitignore | 3 +++ Dockerfile | 8 ++++++++ README.md | 7 +++++++ 3 files changed, 18 insertions(+) create mode 100644 .gitignore create mode 100644 Dockerfile diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..9c88836 --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +data/ +*.pyc +*.log \ No newline at end of file diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..ff1d324 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,8 @@ +FROM pytorch/pytorch:1.3-cuda10.1-cudnn7-devel + +RUN apt update +RUN apt install -y wget libglib2.0-0 libsm6 libxext6 libxrender1 + +COPY requirements.txt /requrements.txt +RUN pip install -r /requrements.txt + diff --git a/README.md b/README.md index 6f87563..660cab8 100644 --- a/README.md +++ b/README.md @@ -24,6 +24,13 @@ pip install -r requirements.txt conda deactivate ``` +### Docker Env with [Nvidia-docker](https://github.com/NVIDIA/nvidia-docker) +```bash +docker build -t asen_img . +docker run --rm --gpus all -v $(pwd)/:/asen -ti asen_img /bin/bash +cd /asen; # inside container +``` + ### Download Data #### Dataset Split