使用rembg移除图片背景失败,加载tensorrt.dll出错
解决rembg运行时加载onnxruntime_providers_tensorrt.dll失败的问题
出现的错误是因为onnxruntime尝试加载TensorRT的动态链接库失败,错误码126一般是缺少依赖、未安装TensorRT或者版本不兼容导致的,给你几个解决办法:
强制用CPU运行,绕开TensorRT问题
修改代码,创建会话时指定只用CPU执行器:from rembg import remove, new_session from PIL import Image input_path = "crop.jpeg" output_path = 'crop1.png' session = new_session("u2net", providers=['CPUExecutionProvider']) input = Image.open(input_path) output = remove(input, session=session) output.save(output_path)安装匹配版本的TensorRT(仅想用GPU加速时需要)
先查看当前onnxruntime-gpu的版本:pip show onnxruntime-gpu根据该版本去下载对应匹配的TensorRT,安装后将TensorRT的lib目录添加到系统PATH环境变量中。
重新安装onnxruntime
如果GPU版onnxruntime存在兼容问题,卸载后换成CPU版本:pip uninstall onnxruntime-gpu pip install onnxruntime
内容的提问来源于stack exchange,提问作者Monisha Sellamuthu
相关产品推荐
相关产品推荐

