diff --git a/.github/workflows/image-pr.yml b/.github/workflows/image-pr.yml index 3df5cd5a..85012135 100644 --- a/.github/workflows/image-pr.yml +++ b/.github/workflows/image-pr.yml @@ -61,7 +61,7 @@ jobs: tag-suffix: '-hipblas' ffmpeg: 'false' image-type: 'extras' - base-image: "rocm/dev-ubuntu-22.04:6.0-complete" + base-image: "rocm/dev-ubuntu-22.04:6.1" grpc-base-image: "ubuntu:22.04" runs-on: 'arc-runner-set' makeflags: "--jobs=3 --output-sync=target" diff --git a/.github/workflows/image.yml b/.github/workflows/image.yml index 43b7052e..3273d36b 100644 --- a/.github/workflows/image.yml +++ b/.github/workflows/image.yml @@ -129,7 +129,7 @@ jobs: ffmpeg: 'true' image-type: 'extras' aio: "-aio-gpu-hipblas" - base-image: "rocm/dev-ubuntu-22.04:6.0-complete" + base-image: "rocm/dev-ubuntu-22.04:6.1" grpc-base-image: "ubuntu:22.04" latest-image: 'latest-gpu-hipblas' latest-image-aio: 'latest-aio-gpu-hipblas' @@ -141,7 +141,7 @@ jobs: tag-suffix: '-hipblas' ffmpeg: 'false' image-type: 'extras' - base-image: "rocm/dev-ubuntu-22.04:6.0-complete" + base-image: "rocm/dev-ubuntu-22.04:6.1" grpc-base-image: "ubuntu:22.04" runs-on: 'arc-runner-set' makeflags: "--jobs=3 --output-sync=target" @@ -218,7 +218,7 @@ jobs: tag-suffix: '-hipblas-ffmpeg-core' ffmpeg: 'true' image-type: 'core' - base-image: "rocm/dev-ubuntu-22.04:6.0-complete" + base-image: "rocm/dev-ubuntu-22.04:6.1" grpc-base-image: "ubuntu:22.04" runs-on: 'arc-runner-set' makeflags: "--jobs=3 --output-sync=target" @@ -228,7 +228,7 @@ jobs: tag-suffix: '-hipblas-core' ffmpeg: 'false' image-type: 'core' - base-image: "rocm/dev-ubuntu-22.04:6.0-complete" + base-image: "rocm/dev-ubuntu-22.04:6.1" grpc-base-image: "ubuntu:22.04" runs-on: 'arc-runner-set' makeflags: "--jobs=3 --output-sync=target" diff --git a/Dockerfile b/Dockerfile index 4f2a73c2..1a036443 100644 --- a/Dockerfile +++ b/Dockerfile @@ -140,6 +140,18 @@ RUN if [ "${BUILD_TYPE}" = "clblas" ]; then \ rm -rf /var/lib/apt/lists/* \ ; fi +RUN if [ "${BUILD_TYPE}" = "hipblas" ]; then \ + apt-get update && \ + apt-get install -y --no-install-recommends \ + hipblas-dev \ + rocblas-dev && \ + apt-get clean && \ + rm -rf /var/lib/apt/lists/* && \ + # I have no idea why, but the ROCM lib packages don't trigger ldconfig after they install, which results in local-ai and others not being able + # to locate the libraries. We run ldconfig ourselves to work around this packaging deficiency + ldconfig \ + ; fi + ################################### ###################################