mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14391 from ROCmSoftwarePlatform:rocm_switch_to_rocm54

Browse Source 
PiperOrigin-RevId: 508497281
This commit is contained in:
jax authors 2023-02-09 15:50:49 -08:00
commit 357b48d29a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
build/rocm/Dockerfile.rocm
View File

@ -1,10 +1,10 @@
FROM ubuntu:focal
MAINTAINER Reza Rahimi <reza.rahimi@amd.com>
MAINTAINER Rahul Batra<rahbatra@amd.com>
ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/5.3/
ARG ROCM_DEB_REPO=http://repo.radeon.com/rocm/apt/5.4/
ARG ROCM_BUILD_NAME=ubuntu
ARG ROCM_BUILD_NUM=main
ARG ROCM_PATH=/opt/rocm-5.3.0
ARG ROCM_PATH=/opt/rocm-5.4.0
ARG DEBIAN_FRONTEND=noninteractive
ENV HOME /root/