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

Switch Dockerfile.rocm to use ROCm 4.0

Browse Source
This commit is contained in:
Deven Desai 2020-12-22 15:43:02 +00:00
parent bba12d0401
commit 3e2f17eafc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/tools/ci_build/Dockerfile.rocm
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.9/
ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/4.0/
ARG ROCM_BUILD_NAME=xenial
ARG ROCM_BUILD_NUM=main
ARG ROCM_PATH=/opt/rocm-3.9.0
ARG ROCM_PATH=/opt/rocm-4.0.0
ENV DEBIAN_FRONTEND noninteractive
ENV TF_NEED_ROCM 1