Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

下载了gpu版本的exe文件,根据onnx1.14安装了CUDA 11.6以及cudnn 8.5.0.69,但是加载segment anything(Vit-b Quant)模型还是报错。 #37

Closed
Burrowsliu opened this issue Jul 28, 2023 · 2 comments

Comments

@Burrowsliu
Copy link

Burrowsliu commented Jul 28, 2023

博主您好:
我下载了gpu版本的exe执行文件,运行加载segment anything模型报错,onnxruntime::python::CreateExecutionProviderInstance CUDA_PATH is set but CUDA wasn't able to be loaded. please install the correct version of CUDA and cuDNN as mentioned in the GPU requirements page,make sure they're in PATH,and that your GPU is supported。
1

以下是我的操作步骤,帮忙看看是什么问题,多谢多谢!!!

1.我检查了CUDA版本,nvcc-V :
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Fri_Dec_17_18:28:54_Pacific_Standard_Time_2021
Cuda compilation tools, release 11.6, V11.6.55
Build cuda_11.6.r11.6/compiler.30794723_0
2
3

2.安装onnx要求的cudnn 8.5.0.96版本,并且将所有lib,include,bin文件拷贝到了CUDA 11.6的文件夹中。

3.进入CUDA安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite运行bandwidthTest.exe以及deviceQuery.exe检测cudnn安装成功。
4
5

4.使用cup版本的exe,运行segment anything (Vit-b Quant)模型没有问题,只是运行gpu版本exe会报错。
6

5.onnxruntime的要注,下载的CUDA与cudn.
8

@Burrowsliu Burrowsliu changed the title 下载了gpu版本的exe文件,根据onnx1.14安装了CUDA 11.6以及cudnn 8.5.0.69,但是加segment anything模型还是报错。 下载了gpu版本的exe文件,根据onnx1.14安装了CUDA 11.6以及cudnn 8.5.0.69,但是加载segment anything(Vit-b Quant)模型还是报错。 Jul 28, 2023
@June1124
Copy link

June1124 commented Aug 2, 2023

我也遇到了同样的问题,测试onnxrunitme输出以下信息
image

@CVHub520
Copy link
Owner

您好,已修复,请尝试 v0.2.0 新版本,后续更新一个Q&A具体介绍下如何自己编译。 @June1124 @Burrowsliu

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants