update rocm to 6.2
This commit is contained in:
parent
b07f7b4ded
commit
dda3c8d03f
@ -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:
|
||||
|
@ -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)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user