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

try to fix dockerfile arch synthax (need testing)

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-29 12:05:53 +02:00
parent f109e9016f
commit 157e6efe16
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
prebuilts/Dockerfile
View File

@ -9,17 +9,18 @@ ENV DEBIAN_FRONTEND=noninteractive \
# For bitsandbytes # For bitsandbytes
ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;gfx1101;gfx1102" ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;gfx1101;gfx1102"
# For LLAMA # For LLAMA
ARG ROCM_DOCKER_ARCH=\ #ARG ROCM_DOCKER_ARCH=\
gfx803 \ # gfx803 \
gfx900 \ # gfx900 \
gfx906 \ # gfx906 \
gfx908 \ # gfx908 \
gfx90a \ # gfx90a \
gfx1010 \ # gfx1010 \
gfx1030 \ # gfx1030 \
gfx1100 \ # gfx1100 \
gfx1101 \ # gfx1101 \
gfx1102 # gfx1102
ARG ROCM_DOCKER_ARCH="gfx803 gfx 900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102"
ENV GPU_TARGETS=${ROCM_DOCKER_ARCH} ENV GPU_TARGETS=${ROCM_DOCKER_ARCH}
ENV CC=/opt/rocm/llvm/bin/clang ENV CC=/opt/rocm/llvm/bin/clang