解决kohya_ss运行时报告Could not load dynamic library ‘libnvinfer.so.7’问题

问题

在运行kohya_ss做lora训练时,会报告
tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library ‘libnvinfer.so.7’; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/nvidia/lib:/usr/local/nvidia/lib64
问题,该如何解决?

解决方案

这个问题并不会影响实际运行,但是每次训练时都会报告这个信息还是比较让人烦的。这个警告是因为tensorflow这个库找不到libnvifer动态库,但是实际上我们并没有真的在使用tensorflow,只是使用了它的tensorboard。如果你不想让这个报错信息出现,可以手工降级tensorflow到2.9.3版本,问题即可解决

发表评论

为防机器,验证码请直接输入4个数字1

*