search

siliconflow / onediff

OneDiff: An out-of-the-box acceleration library for diffusion models.
https://github.com/siliconflow/onediff/wiki
Apache License 2.0
1.41k stars 85 forks source link

Doc Improve: To use onediff in docker pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime you need add LD_LIBRARY_PATH to load libcudnn_cnn_infer.so.8 #896

Closed ganlvtech closed 1 month ago

ganlvtech commented 1 month ago

To use onediff in docker pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime you need add LD_LIBRARY_PATH to load libcudnn_cnn_infer.so.8

export LD_LIBRARY_PATH=/opt/conda/pkgs/pytorch-2.2.2-py3.10_cuda12.1_cudnn8.9.2_0/lib/python3.10/site-packages/torch/lib:$LD_LIBRARY_PATH

https://github.com/siliconflow/onediff/blob/ba93c5a68607abefd38ffed9e6a17bed48c01a81/onediff_comfy_nodes/utils/import_utils.py#L7-L18

importlib.util.find_spec(module_name) is OK. But importlib.import_module(module_name) will fail.

ImportError: libcudnn_cnn_infer.so.8: cannot open shared object file: No such file or directory