YanWenKun / ComfyUI-Docker

🐳Dockerfile for 🎨ComfyUI. | 容器镜像与启动脚本
https://hub.docker.com/r/yanwk/comfyui-boot
Other
467 stars 87 forks source link

fix(LD_LIBRARY_PATH): Add additional path for cudnn #21

Closed M1kep closed 9 months ago

M1kep commented 9 months ago

Fixes following output in container logs:

  0%|          | 0/20 [00:00<?, ?it/s]/usr/lib64/python3.11/site-packages/torch/nn/modules/conv.py:456: UserWarning: Attempt to open cnn_infer failed: handle=0 error: libcudnn_cnn_infer.so.8: cannot open shared object file: No such file or directory (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:78.)
YanWenKun commented 9 months ago

Thank you!