Python version missmatch when installing magenta-realtime/[gpu]

[Legnus]

First of all, loved this project idea! Thank you so much for putting the effort in delivering this.

I'm not a big fan of downloading prepackaged docker containers, so i tried to build the Dockerfile you provided, found in MusicContainer directory. Unfortunately, i got the following error ERROR: Package 'flax' requires a different Python: 3.10.12 not in '>=3.11' on

RUN git clone https://github.com/magenta/magenta-realtime.git && \
    sed -i "s/DEFAULT_SOURCE = 'gcp'/DEFAULT_SOURCE = 'hf'/" magenta-realtime/magenta_rt/asset.py && \
    pip install -e magenta-realtime/[gpu] 

Specifically when installing magenta-realtime/[gpu]

I updated the python system version, in the Dockerfile, to 3.11, along with venv, dev and distutils and it worked out for me.

# System packages
RUN apt update && apt install -y \
  python3.11 python3.11-venv python3.11-dev python3.11-distutils \
  build-essential cmake git curl wget \
  libffi-dev libssl-dev libsndfile1 ffmpeg \
  && rm -rf /var/lib/apt/lists/*

# Set up Python 3.11 as default
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1

Thanks for the detailed report—and for trying to build from the Dockerfile. The error happens because recent Flax/JAX releases now require Python 3.11, so building on Python 3.10 fails when magenta-realtime[gpu] pulls those versions. Two paths: (A) recommended—bump the image to Python 3.11 and install into a 3.11 virtual environment before running the magenta-realtime install (keeping your sed tweak that switches the default asset source to HF). Concretely: install python3.11, python3.11-venv, python3.11-dev, upgrade pip, create a venv, export its bin to PATH, then pip install -e "magenta-realtime[gpu]". (B) If you must stay on Python 3.10, pin JAX to the last release that still supported it (e.g., jax==0.4.31 and jaxlib==0.4.31) and install magenta-realtime[gpu] under those constraints. I can put together a small PR that applies option A to MusicContainer/Dockerfile if that’s helpful. (i told chatgpt to answer for Laurie lol )