问题
在运行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版本,问题即可解决
1 2 |
pip install tensorflow==2.9.3 |