acme icon indicating copy to clipboard operation
acme copied to clipboard

Notebook opened in colab cannot run successfully

Open matchyc opened this issue 5 months ago • 2 comments

I think the quickstart and tutorial notebook will get stuck at the very beginning.

matchyc avatar Aug 05 '25 02:08 matchyc

Can you mention more details either any short video or screenshots!

PredictiveManish avatar Oct 19 '25 09:10 PredictiveManish

This is likely a dependency version mismatch issue that has affected multiple users. Here are potential workarounds:

Option 1: Update numpy first

%pip install --upgrade numpy

Then restart the runtime and re-run.

Option 2: Install with legacy resolver

pip install jaxlib==0.4.20
pip install git+https://github.com/deepmind/acme.git#egg=dm-acme[tf,envs] --use-deprecated=legacy-resolver

Option 3: Specify compatible versions The Acme package has known issues with JAX version compatibility (see #321, #317). You may need to manually pin versions:

jax>=0.4.13
jaxlib>=0.4.13  
chex==0.1.7

Note: Acme's dependencies have not been updated to support the latest JAX versions, which causes these Colab errors. Until the package dependencies are updated, installing from git with pinned versions is the most reliable workaround.

Could you share a screenshot or more specific error details to help narrow down the exact issue?

AryanVBW avatar Dec 03 '25 22:12 AryanVBW