Solution to issue cannot be found in the documentation.
[X] I checked the documentation.
Issue
With this test program and /dev/nvidia* being present in the system
import jax.numpy as np
np.sin(3)
fails with
jaxlib.xla_extension.XlaRuntimeError: INTERNAL: libdevice not found at ./libdevice.10.bc
jax fails to find libdevice
Testing with strace
strace -f python test.py |& grep libdevice
[pid 13910] access("/home/wittgen/lsst/v28/lsstsw/miniconda/envs/lsst-scipipe-9.0.0/lib/python3.11/site-packages/nvidia/cuda_nvcc/nvvm/libdevice", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] access("/home/conda/feedstock_root/build_artifacts/jaxlib_1725045424189/_build_env/targets/x86_64-linux/nvvm/libdevice", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] access("/usr/local/cuda/nvvm/libdevice", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] access("/home/wittgen/lsst/v28/lsstsw/miniconda/envs/lsst-scipipe-9.0.0/lib/python3.11/site-packages/nvidia/cuda_nvcc/nvvm/libdevice", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] access("/home/wittgen/lsst/v28/lsstsw/miniconda/envs/lsst-scipipe-9.0.0/lib/python3.11/nvidia/cuda_nvcc/nvvm/libdevice", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] access("nvvm/libdevice", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] access("libdevice.10.bc", F_OK) = -1 ENOENT (No such file or directory)
[pid 13910] write(2, "jaxlib.xla_extension.XlaRuntimeE"..., 89jaxlib.xla_extension.XlaRuntimeError: INTERNAL: libdevice not found at ./libdevice.10.bc
The default search paths seem wrong. conda installs libdevice for example in
lib/nvvm/libdevice
Solution to issue cannot be found in the documentation.
Issue
With this test program and
/dev/nvidia*
being present in the systemfails with
jax fails to find
libdevice
Testing withstrace
The default search paths seem wrong. conda installs
libdevice
for example inlib/nvvm/libdevice
Installed packages
Environment info