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
|
ARG UBUNTU_VERSION=24.04
|
||||||
|
|
||||||
# This needs to generally match the container host's environment.
|
# This needs to generally match the container host's environment.
|
||||||
ARG ROCM_VERSION=6.3
|
ARG ROCM_VERSION=6.3.4
|
||||||
ARG AMDGPU_VERSION=6.3
|
ARG AMDGPU_VERSION=6.3.4
|
||||||
|
|
||||||
# Target the CUDA build image
|
# 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
|
### Build image
|
||||||
FROM ${BASE_ROCM_DEV_CONTAINER} AS build
|
FROM ${BASE_ROCM_DEV_CONTAINER} AS build
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue