Add requirements.txt

This commit is contained in:
Shangtong Zhang
2018-05-11 13:17:48 -06:00
parent d33f85a2c5
commit aba27ae081
2 changed files with 16 additions and 5 deletions
+4 -5
View File
@@ -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/
+12
View File
@@ -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