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:28:23 +02:00
parent 0bf5f9b84d
commit 12b41b2657
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
2 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -30,13 +30,10 @@ jobs:
# Step 3: Build the Docker image # Step 3: Build the Docker image
- name: Build Docker image - name: Build Docker image
env:
ROCM_VERSION: ${{ inputs.rocm_version }}
TORCH_VERSION: ${{ inputs.torch_version }}
run: | run: |
docker build \ docker build \
--build-arg ROCM_VERSION=${ROCM_VERSION} \ --build-arg ROCM_VERSION=${{ inputs.rocm_version }} \
--build-arg TORCH_VERSION=${TORCH_VERSION} \ --build-arg TORCH_VERSION=${{ inputs.torch_version }} \
-t prebuilt-wheels ./prebuilts -t prebuilt-wheels ./prebuilts
# Step 4: Create a container and run the script # Step 4: Create a container and run the script

5
prebuilts/Dockerfile
View File

@ -1,8 +1,3 @@
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 \