Crash when loading SAM Model

[dhunjoshi]

Using GTX1650 GPU.

Traceback (most recent call last):
File "D:\anaconda3\envs\anylabeling\lib\site-packages\anylabeling\utils.py", line 15, in run
self.func(*self.args, **self.kwargs)
File "D:\anaconda3\envs\anylabeling\lib\site-packages\anylabeling\services\auto_labeling\model_manager.py", line 151, in _load_model
model_info["model"] = SegmentAnything(
File "D:\anaconda3\envs\anylabeling\lib\site-packages\anylabeling\services\auto_labeling\segment_anything.py", line 74, in init
self.encoder_session = onnxruntime.InferenceSession(
File "D:\anaconda3\envs\anylabeling\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 360, in init
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "D:\anaconda3\envs\anylabeling\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 408, in _create_inference_session
sess.initialize_session(providers, provider_options, disabled_optimizers)
RuntimeError: D:\a_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1106 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "D:\anaconda3\envs\anylabeling\lib\site-packages\onnxruntime\capi\onnxruntime_providers_tensorrt.dll"

[web-ho]

Try reinstalling, as mentioned you are using GPU. So, check if you are installing the correct requirements.txt.

If the error didn't resolve check if the specified .dll file is present. Can also look for more info here

[vietanhdev]

@dhunjoshi Please try the latest version:

• For CPU version:

pip install -U anylabeling

• For GPU version:

pip install -U anylabeling-gpu

Thank you!

[vietanhdev]

Documentation for GPU support: https://anylabeling.com/docs/gpu.

[vietanhdev]

From AnyLabeling v0.2.22, to load custom models:

• Download models from here, extract.
• Load the models with Custom Models feature: https://anylabeling.com/docs/custom-models.