experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[ROCm] Update ROCm CI builds to use ROCm 2.8

Browse Source 
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.
This commit is contained in:
Deven Desai 2019-11-15 17:43:13 +00:00
parent 4ff2916c8c
commit f5f3ef0967
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tools/ci_build/Dockerfile.rocm
View File

@ -3,7 +3,7 @@
FROM ubuntu:xenial
MAINTAINER Jeff Poznanovic <jeffrey.poznanovic@amd.com>
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

3
tensorflow/tools/ci_build/builds/docker_test.sh
View File

@ -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=""

3
tensorflow/tools/ci_build/ci_build.sh
View File

@ -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