onnxruntime-gpu飘红报错怎么解决?

这几天在搭comfyui环境的时候遇到一个棘手的问题,使用insightface的时候,GPU报错,只能用CPU,虽然能够运行,但也降低了运行速度。

报错就是大概这样的,在控制台中是明显的红色:

/onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1193 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError] : 1 : FAIL : Failed to load library libonnxruntime_providers_cuda.so with error: libcudart.so.11.0: cannot open shared object file: No such file or directory

报错有少许不同,就类似这种,不用查也知道是onnxruntime-gpu版本的问题,但一直不知道该是哪个版本,今天看到个解决方法,终于有用了。

pip install --upgrade onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

这样安装的是1.17.1的版本。

原本我是1.17.0的版本,cuda的话应该比较高,尝试了1.16多的版本也不行,如果有这个困扰的小伙伴,可以用这个方式,去找下比较搭配的版本。

祝大家成功赶走bug!

相关推荐

  1. onnxruntime-gpu怎么解决

    2024-04-08 13:10:04       44 阅读
  2. docker创建容器时使用--gpus all解决

    2024-04-08 13:10:04       26 阅读
  3. easyexcel解决

    2024-04-08 13:10:04       59 阅读
  4. Mybatis解决

    2024-04-08 13:10:04       35 阅读
  5. 华纳云:SQL Server中offset使用怎么解决

    2024-04-08 13:10:04       50 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-08 13:10:04       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-08 13:10:04       106 阅读
  3. 在Django里面运行非项目文件

    2024-04-08 13:10:04       87 阅读
  4. Python语言-面向对象

    2024-04-08 13:10:04       96 阅读

热门阅读

  1. SpringBoot生成一维码和二维码

    2024-04-08 13:10:04       36 阅读
  2. 系统架构评估_1.相关概念

    2024-04-08 13:10:04       39 阅读
  3. 【flask快速上手(一)】

    2024-04-08 13:10:04       40 阅读
  4. 深入浅出 -- 系统架构之负载均衡Nginx配置SSL证书

    2024-04-08 13:10:04       38 阅读
  5. 使用数据增强和dropout的图像分类

    2024-04-08 13:10:04       27 阅读
  6. 【css】backgroud-position控制背景图位置

    2024-04-08 13:10:04       32 阅读
  7. windows 环境下docker打包python项目

    2024-04-08 13:10:04       41 阅读