update rocm to 6.2

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

View File

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

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}
# also works without -complete for specific arch (tested on gfx1030)