From 4b76a49a1a5741dece6d368b30f7125e20c12878 Mon Sep 17 00:00:00 2001 From: Deven Desai Date: Wed, 26 Aug 2020 15:21:31 +0000 Subject: [PATCH] Updating Dockerfile.rocm to use ROCm 3.7 --- tensorflow/tools/ci_build/Dockerfile.rocm | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tensorflow/tools/ci_build/Dockerfile.rocm b/tensorflow/tools/ci_build/Dockerfile.rocm index 4f5d3ae7291..d209173258a 100644 --- a/tensorflow/tools/ci_build/Dockerfile.rocm +++ b/tensorflow/tools/ci_build/Dockerfile.rocm @@ -3,8 +3,10 @@ FROM ubuntu:bionic MAINTAINER Jeff Poznanovic -ARG DEB_ROCM_REPO=http://repo.radeon.com/rocm/apt/3.3/ -ARG ROCM_PATH=/opt/rocm-3.3.0 +ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/3.7/ +ARG ROCM_BUILD_NAME=xenial +ARG ROCM_BUILD_NUM=main +ARG ROCM_PATH=/opt/rocm-3.7.0 ENV DEBIAN_FRONTEND noninteractive ENV TF_NEED_ROCM 1 @@ -13,8 +15,12 @@ RUN apt update && apt install -y wget software-properties-common # Add rocm repository RUN apt-get clean all -RUN wget -qO - $DEB_ROCM_REPO/rocm.gpg.key | apt-key add - -RUN sh -c "echo deb [arch=amd64] $DEB_ROCM_REPO xenial main > /etc/apt/sources.list.d/rocm.list" +RUN bin/bash -c 'if [[ $ROCM_DEB_REPO == http://repo.radeon.com/rocm/* ]] ; then \ + wget -qO - $ROCM_DEB_REPO/rocm.gpg.key | apt-key add -; \ + echo "deb [arch=amd64] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list; \ + else \ + echo "deb [arch=amd64 trusted=yes] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list ; \ + fi' # Install misc pkgs RUN apt-get update --allow-insecure-repositories && DEBIAN_FRONTEND=noninteractive apt-get install -y \