您的位置:首页 > 其它

ONNX—Anaconda安装

2018-03-29 22:19 836 查看
    今天根据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",不报错即成功安装。


                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  ONNX Anaconda