mirror of
https://github.com/wassname/DeepRL.git
synced 2026-09-09 11:13:47 +08:00
Add requirements.txt
This commit is contained in:
+4
-5
@@ -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/
|
||||
|
||||
@@ -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
|
||||
Reference in New Issue
Block a user