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

try to fix env to workflow

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

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

@ -41,14 +41,8 @@ jobs:
# Step 4: Create a container and run the script # Step 4: Create a container and run the script
- name: Run Docker container and generate wheels - name: Run Docker container and generate wheels
env:
ROCM_VERSION: ${{ inputs.rocm_version }}
TORCH_VERSION: ${{ inputs.torch_version }}
run: | run: |
docker create --name prebuilt-container \ docker create --name prebuilt-container prebuilt-wheels
-e ROCM_ARCH=${ROCM_ARCH} \
-e TORCH_VERSION=${TORCH_VERSION} \
prebuilt-wheels
docker start -a prebuilt-container docker start -a prebuilt-container
# Step 5: Copy bitsandbytes wheel artifact to host # Step 5: Copy bitsandbytes wheel artifact to host

6
prebuilts/Dockerfile
View File

@ -1,3 +1,8 @@
ARG ROCM_VERSION
ARG TORCH_VERSION
ENV ROCM_VERSION=${ROCM_VERSION}
ENV TORCH_VERSION=${TORCH_VERSION}
FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION} FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}
ENV DEBIAN_FRONTEND=noninteractive \ ENV DEBIAN_FRONTEND=noninteractive \
@ -6,7 +11,6 @@ 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
ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102" ENV GPU_TARGETS="gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102"