Open mwittgen opened 1 week ago
@conda-forge/core @jakirkham Do you all have any ideas here?
So libdevice
comes from nvvm
Originally cuda-nvcc-tools
included nvvm
. This works for CUDA 12.0+. However this includes a few extra things that one may not want/need
In CUDA 12.2, cuda-nvvm-tools
was created containing just nvvm
, which cuda-nvcc-tools
depends on. So would recommend installing cuda-nvvm-tools
when using CUDA 12.2+
If these are hard dependencies, the recipe here could be updated to include these. In the interim, users could consider installing one of these packages themselves based on their needs
Hope that helps. Please feel free to ask more questions 🙂
cc @conda-forge/cuda (for vis)
cuda-nvvm-tools 12.6.68 he02047a_0 conda-forge
are installed in our environment, but the default paths
to find libdevice
look in the wrong place. I can fix the setup by creating a symlink or copying into one of the search paths determined by running strace
on the example.
Ah thanks for pointing that out. See that above
Also noticing the path the package uses appears comes from the recipes build
environment instead of the host
environment, which indicates there is a bug in the build script
Looks like the relevant bits are in jaxlib
so will transfer this issue
Yeah think this just needs cuda-nvcc-tools
in requirements/host
. Not seeing it there (though please let me know if I just missed it)
Also think this should be cuda-nvcc-tools
As a side note, think we can drop this workaround as well. It was fixed upstream: https://github.com/conda-forge/cuda-cupti-feedstock/issues/14
I'm including these fixes now in https://github.com/conda-forge/jaxlib-feedstock/pull/281#issuecomment-2378770716
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