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
ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;gfx1101;gfx1102"
# For LLAMA
ARG ROCM_DOCKER_ARCH=\
gfx803 \
gfx900 \
gfx906 \
gfx908 \
gfx90a \
gfx1010 \
gfx1030 \
gfx1100 \
gfx1101 \
gfx1102
#ARG ROCM_DOCKER_ARCH=\
# gfx803 \
# gfx900 \
# gfx906 \
# gfx908 \
# gfx90a \
# gfx1010 \
# gfx1030 \
# gfx1100 \
# gfx1101 \
# gfx1102
ARG ROCM_DOCKER_ARCH="gfx803 gfx 900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102"
ENV GPU_TARGETS=${ROCM_DOCKER_ARCH}
ENV CC=/opt/rocm/llvm/bin/clang