enable gpu

Dockerfile

@@ -6,6 +6,7 @@ ENV PYTHONUNBUFFERED=1
 
 RUN apt-get update && apt-get install --no-install-recommends -y \
   build-essential \
+  python3.11-dev \
   python3.11 \
   python3-pip \
   git \
@@ -38,7 +39,7 @@ WORKDIR $HOME/app
 RUN git clone https://github.com/magenta/magenta-realtime.git && \
     python3.11 -m pip install --no-cache-dir --upgrade gradio && \
     python3.11 -m pip install --no-cache-dir --upgrade pip && \
-    python3.11 -m pip install --no-cache-dir -e magenta-realtime/ && \
+    python3.11 -m pip install --no-cache-dir -e magenta-realtime/[gpu] && \
     python3.11 -m pip uninstall --no-cache-dir -y tensorflow tf-nightly tensorflow-cpu tf-nightly-cpu tensorflow-tpu tf-nightly-tpu tensorflow-hub tf-hub-nightly tensorflow-text tensorflow-text-nightly && \
     python3.11 -m pip install --no-cache-dir tf-nightly==2.20.0.dev20250619 tensorflow-text-nightly==2.20.0.dev20250316 tf-hub-nightly