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",不报错即成功安装。




  • 4
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值