mirror of
https://github.com/ROCm/jax.git
synced 2025-04-15 19:36:06 +00:00
Update Dockerfile.ms
update to ROCm5.4
This commit is contained in:
parent
a2c9fc02e4
commit
0dde7a0fb1
@ -1,8 +1,8 @@
|
||||
################################################################################
|
||||
FROM rocm/dev-ubuntu-20.04:5.3-complete as rt_build
|
||||
MAINTAINER Chao Chen <Chao.Chen.2@amd.com>
|
||||
FROM rocm/dev-ubuntu-20.04:5.4-complete as rt_build
|
||||
MAINTAINER Rahul Batra<rahbatra@amd.com>
|
||||
################################################################################
|
||||
ARG ROCM_PATH=/opt/rocm-5.3.0
|
||||
ARG ROCM_PATH=/opt/rocm-5.4.0
|
||||
|
||||
ARG DEBIAN_FRONTEND=noninteractive
|
||||
ENV HOME /root/
|
||||
|
Loading…
x
Reference in New Issue
Block a user