NM512 / dreamerv3-torch

Implementation of Dreamer v3 in pytorch.
MIT License
422 stars 96 forks source link

Example from README not working #27

Closed StuartShanks closed 1 year ago

StuartShanks commented 1 year ago

Hi, I tried to train the example on DMC Vision but got the following error:

python3 dreamer.py --configs dmc_vision --task dmc_walker_walk --logdir ./logdir/dmc_walker_walk Logdir logdir/dmc_walker_walk Create envs. libEGL warning: MESA-LOADER: failed to open swrast: /usr/lib/dri/swrast_dri.so: cannot open shared object file: No such file or directory (search paths /usr/lib/x86_64-linux-gnu/dri:\$${ORIGIN}/dri:/usr/lib/dri, suffix _dri)

libEGL warning: MESA-LOADER: failed to open swrast: /usr/lib/dri/swrast_dri.so: cannot open shared object file: No such file or directory (search paths /usr/lib/x86_64-linux-gnu/dri:\$${ORIGIN}/dri:/usr/lib/dri, suffix _dri)

Traceback (most recent call last): File "/home/stuart/Github/dreamerv3-torch/dreamer.py", line 418, in main(parser.parse_args(remaining)) File "/home/stuart/Github/dreamerv3-torch/dreamer.py", line 332, in main trainenvs = [make("train") for in range(config.envs)] File "/home/stuart/Github/dreamerv3-torch/dreamer.py", line 332, in trainenvs = [make("train") for in range(config.envs)] File "/home/stuart/Github/dreamerv3-torch/dreamer.py", line 331, in make = lambda mode: make_env(config, logger, mode, train_eps, eval_eps) File "/home/stuart/Github/dreamerv3-torch/dreamer.py", line 190, in make_env env = dmc.DeepMindControl(task, config.action_repeat, config.size) File "/home/stuart/Github/dreamerv3-torch/envs/dmc.py", line 11, in init from dm_control import suite File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/suite/init.py", line 24, in from dm_control.suite import acrobot File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/suite/acrobot.py", line 20, in from dm_control import mujoco File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/mujoco/init.py", line 18, in from dm_control.mujoco.engine import action_spec File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/mujoco/engine.py", line 41, in from dm_control import _render File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/_render/init.py", line 86, in Renderer = import_func() File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/_render/init.py", line 36, in _import_egl from dm_control._render.pyopengl.egl_renderer import EGLContext File "/home/stuart/miniconda3/envs/dreamer3.9/lib/python3.9/site-packages/dm_control/_render/pyopengl/egl_renderer.py", line 75, in raise ImportError('Cannot initialize a headless EGL display.') ImportError: Cannot initialize a headless EGL display.

NM512 commented 1 year ago

@StuartShanks

Hello,

It seems like the error is related to the rendering of DMC (DeepMind Control). You might find a solution by referring to this issue on GitHub. It could be beneficial in resolving the problem you encountered.