[Issue]: rocm/jax images have incorrect PATH

[GMNGeoffrey]

The rocm/jax images rocm/jax:rocm7.0-jax0.6.0-py3.10 and rocm/jax:rocm7.0-jax0.6.0-py3.12 have PATH pointing to the non-existing /opt/rocm-7.0/bin directory

$ docker run --rm -it rocm/jax:rocm7.0-jax0.6.0-py3.10 bash -l
root@8b3139718474:/# echo $PATH
/pyenv/shims:/pyenv/bin:/root/bin:/root/.local/bin:/opt/rocm-7.0/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
root@8b3139718474:/# ls /opt/rocm-7.0/bin
ls: cannot access '/opt/rocm-7.0/bin': No such file or directory
root@8b3139718474:/# ls /opt/
rocm  rocm-7.0.0

$ docker run --rm -it rocm/jax:rocm7.0-jax0.6.0-py3.12 bash -l
root@c0eb34b58fd6:/# echo $PATH
/pyenv/shims:/pyenv/bin:/root/bin:/root/.local/bin:/opt/rocm-7.0/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
root@c0eb34b58fd6:/# ls /opt/rocm-7.0
ls: cannot access '/opt/rocm-7.0': No such file or directory
root@c0eb34b58fd6:/# ls /opt
rocm  rocm-7.0.0

[GMNGeoffrey]

Ping on this? @charleshofer can you PTAL

[GMNGeoffrey]

This appears to have been fixed in rocm/jax:rocm7.1-jax0.7.1-py3.12 but is broken again in rocm/jax:rocm7.2-jax0.8.0-py3.12:

root@mi300x8-001:/app/alphafold# echo $PATH
/root/bin:/root/.local/bin:/opt/rocm-7.2/opencl/bin:/opt/rocm-7.2/bin:/opt/rocm-7.2/hcc/bin:/opt/rocm-7.2//bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/rocm/bin:/opt/rocm/llvm/bin
root@mi300x8-001:/app/alphafold# ls /opt/rocm-7.2
ls: cannot access '/opt/rocm-7.2': No such file or directory
root@mi300x8-001:/app/alphafold# ls /opt
hhsuite  rocm  rocm-7.2.0