ACME with MuJoco

[AbhayGoyal]

Hi,
I am currently working on using ACME, but when I install ACME, I always get issues with Mujoco. I do not need Mujoco at all in my work, as I am not doing any physics-based simulations whatsoever. Could anyone please tell me what should be done in this case?

[AryanVBW]

Acme works well with MuJoCo environments through dm_control.

Setup:

from acme import wrappers
from dm_control import suite

# Load a dm_control environment
env = suite.load('cartpole', 'swingup')

# Wrap for Acme (converts dm_control to dm_env interface)
env = wrappers.SinglePrecisionWrapper(env)

# Use with any continuous-action agent (D4PG, SAC, etc.)
from acme.agents.jax import d4pg
agent = d4pg.D4PG(
    environment_spec=specs.make_environment_spec(env),
    # ... config
)

Common MuJoCo tasks in dm_control:

• suite.load('cartpole', 'swingup')
• suite.load('cheetah', 'run')
• suite.load('humanoid', 'walk')

Note on dependencies: Make sure you have MuJoCo properly installed. See https://github.com/deepmind/dm_control for setup instructions.

Is there a specific MuJoCo task or issue you're encountering?