Update rocm build
Some checks failed
EditorConfig Checker / editorconfig (push) Has been cancelled
Some checks failed
EditorConfig Checker / editorconfig (push) Has been cancelled
This commit is contained in:
parent
3a9457df96
commit
ccc4c16970
1 changed files with 3 additions and 3 deletions
|
@ -1,11 +1,11 @@
|
|||
ARG UBUNTU_VERSION=24.04
|
||||
|
||||
# This needs to generally match the container host's environment.
|
||||
ARG ROCM_VERSION=6.3
|
||||
ARG AMDGPU_VERSION=6.3
|
||||
ARG ROCM_VERSION=6.3.4
|
||||
ARG AMDGPU_VERSION=6.3.4
|
||||
|
||||
# Target the CUDA build image
|
||||
ARG BASE_ROCM_DEV_CONTAINER=rocm/dev-ubuntu-${UBUNTU_VERSION}:${ROCM_VERSION}-complete
|
||||
ARG BASE_ROCM_DEV_CONTAINER=docker.io/rocm/dev-ubuntu-${UBUNTU_VERSION}:${ROCM_VERSION}-complete
|
||||
|
||||
### Build image
|
||||
FROM ${BASE_ROCM_DEV_CONTAINER} AS build
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue