ubuntu安装配置onnxruntime

1.onnxruntime下载配置

项目链接:https://github.com/microsoft/onnxruntime

环境配置链接:Linux 下onnxruntime源码安装

2.cmake安装

下载路径:Download | CMake

环境配置链接:Ubuntu 18.04下安装最新CMake及CMake简单使用

3.python3.7安装

环境配置链接:

ubuntu安装python3.7,并更新python默认指向为python3.7

Python library not found: libpython3.7m.so, libpython3.7.so.1

4.vscode配置

环境配置链接:使用 VSCode 远程开发调试

5.docker配置

带有debug功能的docker run命令

docker run --privileged -it --name name_data_debug -v /home/name/:/home/name image_id /bin/bash

docker exec命令

docker exec -ti container_id bash

环境配置链接:docker debug

6.ps

可能遇到需要配置protobuf的情况

环境配置链接:python 处理protobuf协议

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
可以按照以下步骤安装onnxruntime-gpu: 1. 确保你的GPU支持CUDA和cuDNN,并且已经安装了CUDA和cuDNN。 2. 安装依赖项: ``` sudo apt-get update sudo apt-get install -y git cmake build-essential libprotobuf-dev protobuf-compiler libprotobuf-dev libprotoc-dev libopencv-dev ``` 3. 克隆onnxruntime-gpu仓库: ``` git clone --recursive https://github.com/microsoft/onnxruntime.git ``` 4. 进入onnxruntime目录: ``` cd onnxruntime ``` 5. 创建build目录并进入: ``` mkdir build cd build ``` 6. 配置编译选项: ``` cmake .. -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=/usr/local -Donnxruntime_USE_CUDA=ON -Donnxruntime_CUDA_HOME=/usr/local/cuda -Donnxruntime_CUDA_INCLUDE_DIRS=/usr/local/cuda/include -Donnxruntime_CUDA_LIB_DIRS=/usr/local/cuda/lib64 -Donnxruntime_USE_TENSORRT=OFF ``` 其中,-Donnxruntime_USE_CUDA=ON表示使用CUDA,-Donnxruntime_CUDA_HOME、-Donnxruntime_CUDA_INCLUDE_DIRS、-Donnxruntime_CUDA_LIB_DIRS分别指定CUDA的安装路径和头文件路径、库文件路径。 7. 编译并安装: ``` make -j$(nproc) sudo make install ``` 这里使用了make -j$(nproc)来并行编译,可以加快编译速度。 8. 安装完毕后,可以测试一下onnxruntime-gpu是否正常工作: ``` cd ../tools/ci_build/github/linux/docker/scripts ./run_onnx_tests.sh -e gpu ``` 如果测试通过,则说明onnxruntime-gpu已经安装成功了。 注意:onnxruntime-gpu的安装可能会因为各种原因失败,需要根据具体错误信息进行调试。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值