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

Updating Dockerfile.rocm to use ROCm 3.9

Browse Source
This commit is contained in:
Deven Desai 2020-12-04 02:16:01 +00:00
parent ab0402bd18
commit 0fd4fc2107
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.7/
ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/3.9/
ARG ROCM_BUILD_NAME=xenial
ARG ROCM_BUILD_NUM=main
ARG ROCM_PATH=/opt/rocm-3.7.0
ARG ROCM_PATH=/opt/rocm-3.9.0
ENV DEBIAN_FRONTEND noninteractive
ENV TF_NEED_ROCM 1