diff --git a/build/rocm/Dockerfile.rocm b/build/rocm/Dockerfile.rocm index d0d801806d71..67047d7f7586 100644 --- a/build/rocm/Dockerfile.rocm +++ b/build/rocm/Dockerfile.rocm @@ -1,10 +1,10 @@ FROM ubuntu:focal -MAINTAINER Reza Rahimi +MAINTAINER Rahul Batra -ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/5.3/ +ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/5.4/ ARG ROCM_BUILD_NAME=ubuntu ARG ROCM_BUILD_NUM=main -ARG ROCM_PATH=/opt/rocm-5.3.0 +ARG ROCM_PATH=/opt/rocm-5.4.0 ARG DEBIAN_FRONTEND=noninteractive ENV HOME /root/