diff --git a/build_rocm_python3 b/build_rocm_python3 index 0c88adc8351134..e7d6cbdc92e647 100755 --- a/build_rocm_python3 +++ b/build_rocm_python3 @@ -26,7 +26,7 @@ done shift "$((OPTIND-1))" # First positional argument (if any) specifies the ROCM_INSTALL_DIR -ROCM_INSTALL_DIR=/opt/rocm-6.1.2 +ROCM_INSTALL_DIR=/opt/rocm/ if [[ -n $1 ]]; then ROCM_INSTALL_DIR=$1 fi diff --git a/rocm_docs/tensorflow-build-from-source.md b/rocm_docs/tensorflow-build-from-source.md index d4e026d43b3c6f..e78b29a45c1f0b 100644 --- a/rocm_docs/tensorflow-build-from-source.md +++ b/rocm_docs/tensorflow-build-from-source.md @@ -13,7 +13,7 @@ For details of the ROCm instructions, please refer to the [ROCm QuickStart Insta To build with ROCm3.10, set the following environment variables, and add those environment variables at the end of ~/.bashrc ``` -export ROCM_PATH=/opt/rocm-6.1.2 +export ROCM_PATH=/opt/rocm/ export HCC_HOME=$ROCM_PATH/hcc export HIP_PATH=$ROCM_PATH/hip export PATH=$HCC_HOME/bin:$HIP_PATH/bin:$PATH diff --git a/tensorflow/tools/ci_build/Dockerfile.rocm b/tensorflow/tools/ci_build/Dockerfile.rocm index d834b14808f88f..8d188d5a34bdcb 100644 --- a/tensorflow/tools/ci_build/Dockerfile.rocm +++ b/tensorflow/tools/ci_build/Dockerfile.rocm @@ -6,7 +6,7 @@ MAINTAINER Jeff Poznanovic ARG ROCM_DEB_REPO=https://repo.radeon.com/rocm/apt/6.1.2/ ARG ROCM_BUILD_NAME=ubuntu ARG ROCM_BUILD_NUM=main -ARG ROCM_PATH=/opt/rocm-6.1.2 +ARG ROCM_PATH=/opt/rocm/ ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100" ARG DEBIAN_FRONTEND=noninteractive diff --git a/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh b/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh index 72eb5430e795ee..539be5027daa60 100755 --- a/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh +++ b/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh @@ -29,7 +29,7 @@ if [[ -n $1 ]]; then ROCM_INSTALL_DIR=$1 else if [[ -z "${ROCM_PATH}" ]]; then - ROCM_INSTALL_DIR=/opt/rocm-6.1.2 + ROCM_INSTALL_DIR=/opt/rocm/ else ROCM_INSTALL_DIR=$ROCM_PATH fi diff --git a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh index 50a68cbba27094..b1d486da8f0bcb 100755 --- a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh +++ b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh @@ -37,7 +37,7 @@ if [[ -n $1 ]]; then ROCM_INSTALL_DIR=$1 else if [[ -z "${ROCM_PATH}" ]]; then - ROCM_INSTALL_DIR=/opt/rocm-6.1.2 + ROCM_INSTALL_DIR=/opt/rocm/ else ROCM_INSTALL_DIR=$ROCM_PATH fi