From 8bda0bb4d662970e0e0c024c0b0aa378290542e7 Mon Sep 17 00:00:00 2001 From: sunway513 Date: Tue, 15 Feb 2022 15:25:19 +0000 Subject: [PATCH] updates for rocm5.0 docker --- build_all.sh | 5 ++--- push_all.sh | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/build_all.sh b/build_all.sh index 89ddece..7b207b8 100755 --- a/build_all.sh +++ b/build_all.sh @@ -1,5 +1,5 @@ -ROCM_VERSION=4.5.2 -AMDGPU_VERSION=21.40.2 +ROCM_VERSION=5.0 +AMDGPU_VERSION=21.50 cp -r scripts rocm-terminal cp -r scripts dev @@ -33,4 +33,3 @@ sudo docker build . -f Dockerfile.post-install --build-arg=base_image=rocm/dev-u #ubuntu20.04 complete sudo docker build . -f Dockerfile-ubuntu-20.04-complete -t rocm/dev-ubuntu-20.04:$ROCM_VERSION-complete --no-cache --build-arg=ROCM_VERSION=$ROCM_VERSION --build-arg=AMDGPU_VERSION=$AMDGPU_VERSION sudo docker build . -f Dockerfile.post-install --build-arg=base_image=rocm/dev-ubuntu-20.04:$ROCM_VERSION-complete -t rocm/dev-ubuntu-20.04:$ROCM_VERSION-complete - diff --git a/push_all.sh b/push_all.sh index fe14480..256a8a2 100755 --- a/push_all.sh +++ b/push_all.sh @@ -1,4 +1,4 @@ -ROCM_VERSION=4.5.2 +ROCM_VERSION=5.0 sudo docker push rocm/dev-ubuntu-18.04:$ROCM_VERSION sudo docker push rocm/dev-ubuntu-20.04:$ROCM_VERSION sudo docker push rocm/dev-centos-7:$ROCM_VERSION