我在自己的服务器上build了镜像：

singularity pull --arch amd64 docker://image.sourcefind.cn:5000/dcu/admin/base/vscode-jax:0.4.23-ubuntu20.04-dtk24.04.1-py3.10

目的是想基于JAX在DCU硬件上开发。然而我scp到集群上之后，如下操作测试时遇到了报错：

$ module load singularity/3.8.0
$ singularity shell --bind /var/log/user-log:/var/log/user-log vscode-jax_0.4.23-ubuntu20.04-dtk24.04.1-py3.10.sif
Singularity> python
Python 3.10.8 (main, Nov 24 2022, 14:13:03) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import jax
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/opt/conda/lib/python3.10/site-packages/jax/__init__.py", line 39, in <module>
    from jax import config as _config_module
  File "/opt/conda/lib/python3.10/site-packages/jax/config.py", line 15, in <module>
    from jax._src.config import config as _deprecated_config  # noqa: F401
  File "/opt/conda/lib/python3.10/site-packages/jax/_src/config.py", line 28, in <module>
    from jax._src import lib
  File "/opt/conda/lib/python3.10/site-packages/jax/_src/lib/__init__.py", line 82, in <module>
    import jaxlib.cpu_feature_guard as cpu_feature_guard
ImportError: libhsa-runtime64.so.1: cannot open shared object file: No such file or directory
>>>

请问如何解决呢？我自行搜索了下问题，但是看不太懂。
https://github.com/ROCm/ROCm/issues/1852
https://github.com/ROCm/ROCm/issues/1581