mathieu/ai-suite-rocm-local
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

switch to -complete docker image

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-28 16:02:15 +02:00
parent be6be8a529
commit b280b7fa09
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/build-and_publish-prebuilts-whl.yml vendored
View File

@ -17,8 +17,10 @@ jobs:
runs-on: ubuntu-latest runs-on: ubuntu-latest
steps: steps:
- name: Delete huge unnecessary tools folder - name: Free Disk Space (Ubuntu)
run: rm -rf /opt/hostedtoolcache uses: jlumbroso/free-disk-space@main
with:
tool-cache: false
# Step 1: Checkout the repository # Step 1: Checkout the repository
- name: Checkout repository - name: Checkout repository

5
prebuilts/Dockerfile
View File

@ -1,5 +1,6 @@
ARG ROCM_VERSION=6.1.2 ARG ROCM_VERSION=6.1.2
FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION} FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}-complete
# also works without -complete for specific arch (tested on gfx1030)
ENV DEBIAN_FRONTEND=noninteractive \ ENV DEBIAN_FRONTEND=noninteractive \
PYTHONUNBUFFERED=1 \ PYTHONUNBUFFERED=1 \
@ -10,6 +11,8 @@ ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;
# For LLAMA # For LLAMA
ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101" ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101"
ENV CC=/opt/rocm/llvm/bin/clang
ENV CXX=/opt/rocm/llvm/bin/clang++
WORKDIR /tmp WORKDIR /tmp