diff --git a/Dockerfile.rocm b/Dockerfile.rocm index cdb8034..46706c6 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -17,6 +17,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ WORKDIR /app +# Install uv so packages that use uv-build as their build backend can be built. +RUN pip3 install uv + # Step 1: Install ROCm-compatible PyTorch stack first. # This must happen before anything else to prevent pip from pulling CPU wheels. COPY requirements-rocm-init.txt .