vietanhdev / anylabeling

Effortless AI-assisted data labeling with AI support from YOLO, Segment Anything, MobileSAM!!
https://anylabeling.nrl.ai
GNU General Public License v3.0
2.04k stars 222 forks source link

How should I configure GPU inference using the exe installation package? #86

Open ZJDATY opened 1 year ago

ZJDATY commented 1 year ago

After selecting the model, it will report this error. image

My local environment is cuda11.7+cudnn8.6+onnxruntime-gpu1.14.0. All of the above have added environment variables. CUDA_ PATH is also configured correctly. I have also placed the DLL file for onnxruntime-gpu1.14.0 in the running directory. image The model using YOLOV8N did not report any errors, but the task manager did not use GPU inference. @vietanhdev May I ask what I should do?

vietanhdev commented 1 year ago

Please check the version of CUDA and CUDNN:

ZJDATY commented 1 year ago

Please check the version of CUDA and CUDNN:

@vietanhdev I have set the path of the environment variable as required. Using cuda11.6.2 and cudnn 8.5.0.96. And copied all the folders of cudnn to/ Under the path of CUDA/v11.6/. But the software still reported the same error. image image image image image

ZJDATY commented 1 year ago

I placed all the correct version of the DLL files in the exe path, but the same error was still reported.

image

CryMasK commented 1 year ago

I face the same problem.

I used CUDA 11.8 and cudnn 8.9.3 at the first time. It detected a segment successfully, then the program crashed immediately.
When I reopened the program and tried to load the model, it showed the error.

I found this issue ticket, and installed the correct version of CUDA and cudnn, but still no luck.

Loongle commented 5 months ago

@ZJDATY I have same problem, Did you solve it?