diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 137452cad2c15..731a97d93da1f 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -17,10 +17,7 @@ RUN python3 -m pip install --upgrade pip RUN apt-get purge -y sccache; python3 -m pip uninstall -y sccache; rm -f "$(which sccache)" # Install UV -RUN curl -LsSf https://astral.sh/uv/install.sh | sh - -# Activate virtual environment and add uv to PATH -ENV PATH="/root/.local/bin:$PATH" +RUN curl -LsSf https://astral.sh/uv/install.sh | env UV_INSTALL_DIR="/usr/local/bin" sh # This timeout (in seconds) is necessary when installing some dependencies via uv since it's likely to time out # Reference: https://github.com/astral-sh/uv/pull/1694