From aba27ae0815f766144c1d52fa4bc828f843d177a Mon Sep 17 00:00:00 2001
From: Shangtong Zhang <zhangshangtong.cpp@gmail.com>
Date: Fri, 11 May 2018 13:17:48 -0600
Subject: [PATCH] Add requirements.txt

---
 Dockerfile       |  9 ++++-----
 requirements.txt | 12 ++++++++++++
 2 files changed, 16 insertions(+), 5 deletions(-)
 create mode 100644 requirements.txt

diff --git a/Dockerfile b/Dockerfile
index daf5464..e1c1622 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -7,12 +7,11 @@ RUN apt install -y --allow-unauthenticated --no-install-recommends \
         libgtk3.0 libsm6 python3-dev
 RUN rm -rf /var/lib/apt/lists/*
 
-WORKDIR /workspace/
-COPY deep_rl deep_rl
-COPY setup.py setup.py
-COPY examples.py examples.py
+WORKDIR /tmp
 RUN pip3 install --upgrade pip
 RUN ln -s /usr/bin/python3 /usr/bin/python
 RUN ln -s /usr/bin/pip3 /usr/bin/pip
-RUN pip install -e .
+COPY requirements.txt requirements.txt
+RUN pip install -r requirements.txt
 RUN pip install pybullet
+WORKDIR /workspace/
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..61ac659
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,12 @@
+torch>=0.4.0
+torchvision>=0.2.1
+gym>=0.10.5
+atari-py>=0.1.1
+opencv-python>=3.4.0.12
+tensorboardX==1.1
+scikit-image>=0.13.1
+tqdm>=4.23.0
+pandas>=0.22.0
+pathlib2>=1.0.1;python_version <= '2.7'
+pathlib>=1.0.1;python_version >= '3.5'
+seaborn>=0.8.1