Fix Dockerfile for Ubuntu 24.04 (Noble) base image

- Replace python3.10 packages with python3 equivalents
- Use noble ROCm package instead of jammy
- Add --break-system-packages for pip on Noble
- Remove git-core PPA (not needed on Noble)
- Remove linux-headers install (not available during build)

Author: Mark Saroufim

docker/amd-docker.Dockerfile

@@ -3,10 +3,8 @@ FROM ghcr.io/actions/actions-runner:latest
 ENV CXX=clang++
 
 RUN sudo apt-get update -y \
-    && sudo apt-get install -y software-properties-common \
-    && sudo add-apt-repository -y ppa:git-core/ppa \
-    && sudo apt-get update -y \
     && sudo apt-get install -y --no-install-recommends \
+    software-properties-common \
     curl \
     ca-certificates \
     git \
@@ -20,46 +18,47 @@ RUN sudo apt-get update -y \
     lld \
     wget \
     psmisc \
-    python3.10-venv \
+    python3-venv \
+    python3-pip \
+    python3-setuptools \
+    python3-wheel \
+    python3-dev \
     && sudo rm -rf /var/lib/apt/lists/*
 
-RUN sudo apt-get update && sudo apt-get install -y python3.10 python3-pip python-is-python3 python3-setuptools python3-wheel libpython3.10
-
 RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash && \
     sudo apt-get install git-lfs
 
 RUN sudo groupadd -g 109 render
 
 RUN sudo apt update -y \
-    && sudo apt install -y "linux-headers-$(uname -r)" "linux-modules-extra-$(uname -r)" \
     && sudo usermod -a -G render,video runner \
-    && wget https://repo.radeon.com/amdgpu-install/7.1/ubuntu/jammy/amdgpu-install_7.1.70100-1_all.deb \
+    && wget https://repo.radeon.com/amdgpu-install/7.1/ubuntu/noble/amdgpu-install_7.1.70100-1_all.deb \
     && sudo apt install -y ./amdgpu-install_7.1.70100-1_all.deb \
     && sudo apt update -y \
     && sudo apt install -y rocm
 
 ENV ROCM_PATH=/opt/rocm
 
-RUN sudo pip install --upgrade pip
+RUN sudo pip install --break-system-packages --upgrade pip
 
-RUN sudo pip install --no-cache-dir torch==2.10.0+rocm7.1 --index-url https://download.pytorch.org/whl/rocm7.1
+RUN sudo pip install --break-system-packages --no-cache-dir torch==2.10.0+rocm7.1 --index-url https://download.pytorch.org/whl/rocm7.1
 
 RUN git clone --recursive https://github.com/ROCm/aiter.git \
     && cd aiter \
     && git checkout f3be04a12a0cfd6b5e2c7a94edc774f1bc24460d \
-    && sudo pip install -r requirements.txt \
+    && sudo pip install --break-system-packages -r requirements.txt \
     && sudo python3 setup.py develop
 
 RUN sudo mkdir -p /home/runner/aiter/aiter/jit/build \
     && sudo chown -R runner:runner /home/runner/aiter/aiter/jit/build
 
-RUN sudo pip install \
+RUN sudo pip install --break-system-packages \
     ninja \
     numpy \
     packaging \
     wheel \
     tinygrad
 
-RUN sudo pip install git+https://github.com/ROCm/iris.git
+RUN sudo pip install --break-system-packages git+https://github.com/ROCm/iris.git
 
 ENV LD_LIBRARY_PATH="/opt/rocm/lib"