Updating Dockerfile.rocm and ROCm CI scripts to use ROCm 3.9

This commit is contained in:
Deven Desai 2020-10-29 02:14:04 +00:00
parent c43ac22a0c
commit 34ec724850
5 changed files with 6 additions and 6 deletions

View File

@ -3,10 +3,10 @@
FROM ubuntu:bionic
MAINTAINER Jeff Poznanovic <jeffrey.poznanovic@amd.com>
ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/3.7/
ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/3.9/
ARG ROCM_BUILD_NAME=xenial
ARG ROCM_BUILD_NUM=main
ARG ROCM_PATH=/opt/rocm-3.7.0
ARG ROCM_PATH=/opt/rocm-3.9.0
ENV DEBIAN_FRONTEND noninteractive
ENV TF_NEED_ROCM 1

View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi

View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi

View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi

View File

@ -30,7 +30,7 @@ export PYTHON_BIN_PATH=`which python3`
export CC_OPT_FLAGS='-mavx'
export TF_NEED_ROCM=1
export ROCM_PATH=/opt/rocm-3.3.0
export ROCM_PATH=/opt/rocm-3.9.0
export TF_GPU_COUNT=${N_GPUS}
yes "" | $PYTHON_BIN_PATH configure.py