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