Dockerfile changes to build for ROCm

Author: fenfir

Dockerfile

@@ -1,7 +1,8 @@
-ARG GO_VERSION=1.21-bullseye
+ARG GO_VERSION=1.21-bookworm
 ARG IMAGE_TYPE=extras
 # extras or core
 
+FROM rocm/dev-ubuntu-22.04:6.0-complete as rocm
 
 FROM golang:$GO_VERSION as requirements-core
 
@@ -40,6 +41,11 @@ RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
     ; fi
 ENV PATH /usr/local/cuda/bin:${PATH}
 
+# HipBLAS requirements
+COPY --from=rocm /opt/rocm /opt/rocm
+ENV PATH /opt/rocm/bin:${PATH}
+ENV ROCM_PATH=/opt/rocm
+
 # OpenBLAS requirements and stable diffusion
 RUN apt-get install -y \
     libopenblas-dev \
@@ -66,7 +72,7 @@ RUN curl https://repo.anaconda.com/pkgs/misc/gpgkeys/anaconda.asc | gpg --dearmo
     apt-get install -y conda && apt-get clean
 
 ENV PATH="/root/.cargo/bin:${PATH}"
-RUN pip install --upgrade pip
+RUN apt-get install -y python3-pip && apt-get clean
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
 RUN apt-get install -y espeak-ng espeak && apt-get clean