danijar / dreamerv2

Mastering Atari with Discrete World Models
https://danijar.com/dreamerv2
MIT License
886 stars 195 forks source link

Update Dockerfile to use tensorflow/tensorflow:2.10.0-gpu #47

Open ikeyasu opened 1 year ago

ikeyasu commented 1 year ago

Two issues on Dockerfile

tensorflow:2.4.2-gpu has an issue on

RUN apt-get update && apt-get install -y \
  ffmpeg \
  libgl1-mesa-dev \
  python3-pip \
  unrar \
  wget \
  && apt-get clean

A folder structure of http://www.atarimania.com/roms/Roms.rar has been changed. Removing unzip.