[Issue]: libdw.so.1 not found in ubuntu docker images

[simutisernestas]

Problem Description

Python 3.12.12 | packaged by conda-forge | (main, Oct 22 2025, 23:25:55) [GCC 14.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import jax
>>> jax.devices()
ERROR:2025-10-31 12:27:02,228:jax._src.xla_bridge:444: Jax plugin configuration error: Exception when calling jax_plugins.xla_rocm7.initialize()
Traceback (most recent call last):
  File "/tmp/env/lib/python3.12/site-packages/jax/_src/xla_bridge.py", line 442, in discover_pjrt_plugins
    plugin_module.initialize()
  File "/tmp/env/lib/python3.12/site-packages/jax_plugins/xla_rocm7/__init__.py", line 137, in initialize
    c_api = xb.register_plugin(
            ^^^^^^^^^^^^^^^^^^^
  File "/tmp/env/lib/python3.12/site-packages/jax/_src/xla_bridge.py", line 551, in register_plugin
    c_api = xla_client.load_pjrt_plugin_dynamically(plugin_name, library_path)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/tmp/env/lib/python3.12/site-packages/jaxlib/xla_client.py", line 111, in load_pjrt_plugin_dynamically
    return _xla.load_pjrt_plugin(plugin_name, library_path, c_api=None)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
jaxlib._jax.XlaRuntimeError: INTERNAL: Failed to open /tmp/env/lib/python3.12/site-packages/jax_plugins/xla_rocm7/xla_rocm_plugin.so: libdw.so.1: cannot open shared object file: No such file or directory
[CpuDevice(id=0)]

Operating System

Ubuntu 24.04

CPU

not relevant

GPU

not relevant

ROCm Version

rocm-7.0.2

ROCm Component

No response

Steps to Reproduce

docker run -it rocm/dev-ubuntu-24.04:7.0.2-complete bash
ldconfig -p | grep libdw

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response

[simutisernestas]

same issue in rocm/jax:rocm7.0.2-jax0.6.0-py3.12

[lucbruni-amd]

Hi @simutisernestas, thank you for the report. Just wanted to let you know there are some open PRs to address this:

#160
ROCm/ROCm-docker#158

160 is more temporary until 158 is merged. This is expected to be fixed for ROCm 7.1.

As a temporary workaround, please manually install libdw1t64 for Ubuntu 24.04 images and libdw1 for Ubuntu 22.04.

Apologies for the inconvenience, and we appreciate your report. Please let me know if there are further issues. Thanks!