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

Create a new ROCM RBE Docker image with ROCM 4.0.

Browse Source 
Update the hash accordingly.

PiperOrigin-RevId: 350711537
Change-Id: I8ef9a84d59831c377e867bb6c88dd4307953b365
This commit is contained in:
Adrian Kuegel 2021-01-08 00:04:29 -08:00 committed by TensorFlower Gardener
parent ae1b19aaaa
commit 6a848f27ab
2 changed files with 4 additions and 4 deletions
tensorflow/tools/ci_build
Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython
third_party/toolchains/preconfig/generate
containers.bzl

6
tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython
View File

@ -16,8 +16,8 @@ ARG DEBIAN_FRONTEND=noninteractive
# Install ROCm packages
RUN apt-get update && apt-get install -y --no-install-recommends \
curl libnuma-dev gnupg sudo libelf1 build-essential \
&& curl -sL http://repo.radeon.com/rocm/apt/3.9/rocm.gpg.key | apt-key add - \
&& printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/3.9/ xenial main" | tee /etc/apt/sources.list.d/rocm.list \
&& curl -sL http://repo.radeon.com/rocm/apt/4.0/rocm.gpg.key | apt-key add - \
&& printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/4.0/ xenial main" | tee /etc/apt/sources.list.d/rocm.list \
&& apt-get update && apt-get install -y --no-install-recommends \
rocm-dev rocm-libs hipcub rocm-utils rocm-cmake \
rocfft miopen-hip miopengemm rocblas hipblas rocrand rccl \
@ -34,7 +34,7 @@ ENV PATH="$OPENCL_ROOT/bin:${PATH}"
# Set target file to help determine which device(s) to build for
RUN bash -c "ls -al /opt/roc*"
RUN bash -c "echo -e 'gfx803\ngfx900\ngfx906' > $ROCM_PATH/bin/target.lst"
RUN bash -c "echo -e 'gfx803\ngfx900\ngfx906\ngfx908' > $ROCM_PATH/bin/target.lst"
# Copy and run the install scripts.
COPY install/install_bootstrap_deb_packages.sh /install/

2
third_party/toolchains/preconfig/generate/containers.bzl vendored
View File

@ -12,6 +12,6 @@ container_digests = {
"cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython": "sha256:3f890a951c81a201d60d0161a56ce628a90323be0c7f795550caa37f6f41a85c",
"cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython": "sha256:bd7666d1ef49b2b2e2a64981f1c9234deeccdb0d5198b30ff4289c3dfcffedbf",
"cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython": "sha256:f436545b7e14b014393b42975923dcd01f408496b1399abb5a35608f888ca140",
"rocm-ubuntu18.04-manylinux2010-multipython": "sha256:bf6fd9711e82ae2c43229ef97f972db6d35975fd5774da3fb2e64bf952c426d0",
"rocm-ubuntu18.04-manylinux2010-multipython": "sha256:07ebbcd24eb1ec0819c0a57f789808ddd3f293f210efbb9e0525c22dc44729a2",
"windows-1803": "sha256:f109576c7c0c8a1783ff22b666e8923b52dbbe7933f69a1c7a7275202c304a12",
}