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

update rocm to 6.2

Browse Source
This commit is contained in:
Mathieu Broillet 2024-09-23 18:28:48 +02:00
parent b07f7b4ded
commit dda3c8d03f
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
2 changed files with 4 additions and 4 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

@ -6,11 +6,11 @@ on:
rocm_version: rocm_version:
description: 'ROCm version' description: 'ROCm version'
required: true required: true
default: '6.1.2' default: '6.2'
torch_version: torch_version:
description: 'Torch version (e.g., rocm6.1)' description: 'Torch version (e.g., rocm6.2)'
required: true required: true
default: 'rocm6.1' default: 'rocm6.2'
jobs: jobs:
build: build:

2
prebuilts/Dockerfile
View File

@ -1,4 +1,4 @@
ARG ROCM_VERSION=6.1.2 ARG ROCM_VERSION=6.2
FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION} FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}
# also works without -complete for specific arch (tested on gfx1030) # also works without -complete for specific arch (tested on gfx1030)