From f5f3ef09671dabdedf6fdd422dda0d144efbd058 Mon Sep 17 00:00:00 2001 From: Deven Desai Date: Fri, 15 Nov 2019 17:43:13 +0000 Subject: [PATCH] [ROCm] Update ROCm CI builds to use ROCm 2.8 This PR/commit updates the Dockerfile.rocm file to use ROCm version 2.8 (from the current 2.6). Switching to ROCm version 2.8, also adds to the requirement of specifying a couple of extra option to the `docker run` command. That change is also a part of this PR/commit. --- tensorflow/tools/ci_build/Dockerfile.rocm | 2 +- tensorflow/tools/ci_build/builds/docker_test.sh | 3 ++- tensorflow/tools/ci_build/ci_build.sh | 3 ++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tensorflow/tools/ci_build/Dockerfile.rocm b/tensorflow/tools/ci_build/Dockerfile.rocm index 191947da7a7..a083bc6debd 100644 --- a/tensorflow/tools/ci_build/Dockerfile.rocm +++ b/tensorflow/tools/ci_build/Dockerfile.rocm @@ -3,7 +3,7 @@ FROM ubuntu:xenial MAINTAINER Jeff Poznanovic -ARG DEB_ROCM_REPO=http://repo.radeon.com/rocm/apt/2.6/ +ARG DEB_ROCM_REPO=http://repo.radeon.com/rocm/apt/2.8.0/ ARG ROCM_PATH=/opt/rocm ENV DEBIAN_FRONTEND noninteractive diff --git a/tensorflow/tools/ci_build/builds/docker_test.sh b/tensorflow/tools/ci_build/builds/docker_test.sh index 38891b60e57..39e119f8895 100755 --- a/tensorflow/tools/ci_build/builds/docker_test.sh +++ b/tensorflow/tools/ci_build/builds/docker_test.sh @@ -109,7 +109,8 @@ if [ "${IMAGE_TYPE}" == "gpu" ]; then libs=$(\ls /usr/lib/x86_64-linux-gnu/libcuda.* | xargs -I{} echo '-v {}:{}') GPU_EXTRA_PARAMS="${devices} ${libs}" elif [ "${IMAGE_TYPE}" == "rocm" ]; then - ROCM_EXTRA_PARAMS="--device=/dev/kfd --device=/dev/dri --group-add video" + ROCM_EXTRA_PARAMS="--device=/dev/kfd --device=/dev/dri --group-add video \ + --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size 16G" else GPU_EXTRA_PARAMS="" ROCM_EXTRA_PARAMS="" diff --git a/tensorflow/tools/ci_build/ci_build.sh b/tensorflow/tools/ci_build/ci_build.sh index 079765bd5f9..d41972f4e1a 100755 --- a/tensorflow/tools/ci_build/ci_build.sh +++ b/tensorflow/tools/ci_build/ci_build.sh @@ -111,7 +111,8 @@ fi # Add extra params for rocm devices and libraries for ROCm container. if [[ "${CONTAINER_TYPE}" == "rocm" ]]; then - ROCM_EXTRA_PARAMS="--device=/dev/kfd --device=/dev/dri --group-add video" + ROCM_EXTRA_PARAMS="--device=/dev/kfd --device=/dev/dri --group-add video \ + --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size 16G" else ROCM_EXTRA_PARAMS="" fi