您好,欢迎来到爱玩科技网。
搜索
您的当前位置:首页ONNX—Anaconda安装

ONNX—Anaconda安装

来源:爱玩科技网
    今天根据pytorch官网教程配置ONNX,发现教程中还存在一些坑,经过问题分析查找,现已将问题解决,成功安装。具体步骤如下:
    1:创建python3.5环境,将TensorRT和Pytorch安装在此环境中,否则使用python3.6版本TensorRT将无法安装成功;
    2:source activate XXX35(python3.5环境)切换到安装环境,执行
     conda install -c conda-forge protobuf numpy
     conda install -c conda-forge onnx
    3:在此环境下  使用 python -c "import onnx"发现报错:
     Error in `python': free(): invalid pointer
      很可能是缺少libtcmalloc-minimal4库,首先安装
     sudo apt-get install libtcmalloc-minimal4
     然后打开~/.bashrc文件
     sudo gedit ~/.bashrc
     在文件末尾添加如下代码:
     export LD_PRELOAD="/usr/lib/libtcmalloc_minimal.so.4"
     最后重新载入环境变量即可
     sudo source ~/.bashrc
    4:关闭当前terminal并重新打开一个新的terminal,切换进XXX35环境,使用 python -c "import onnx",不报错即成功安装。




因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- aiwanbo.com 版权所有 赣ICP备2024042808号-3

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务