hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added Docker instruction

Browse Source
This commit is contained in:
NM512
2023-10-02 07:37:08 +09:00
parent 1002d8b115
commit 0eec12bd1c
4 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Dockerfile
View File

@@ -6,10 +6,10 @@
#
# 2. Start training:
# docker build -f Dockerfile -t img . && \
# docker run -it --rm --gpus all -v $PWD:/workspace img \
# docker run -it --rm --gpus all -v $PWD:/workspace -u $(id -u):$(id -g) img \
# sh xvfb_run.sh python3 dreamer.py \
# --logdir "./logdir/dmc_walker_walk" \
# --configs dmc_vision --task dmc_walker_walk
# --configs dmc_vision --task dmc_walker_walk \
# --logdir "./logdir/dmc_walker_walk"
#
# 3. See results:
# tensorboard --logdir ~/logdir
@@ -34,11 +34,11 @@ ENV NUMBA_CACHE_DIR=/tmp
# dmc setup
RUN pip3 install tensorboard
RUN pip3 install gym==0.19.0
RUN pip3 install dm_control
RUN pip3 install dm_control==1.0.9
RUN pip3 install moviepy
# crafter setup
RUN pip3 install crafter
RUN pip3 install crafter==1.8.0
# atari setup
RUN pip3 install atari-py==0.2.9