环境配置
需要基于PyTorch1.1以上的版本,并安装onnx库:
pip install onnx
工具包安装
方法1:pip 安装
pip install easypruner
方法2 源码+环境变量
git clone https://ivapruner:qweruiop123@gitee.com/casia_iva_engineer/pruning-tools.git
将工具包添加进python的import路径即可: 其中,$(EasyPrunerPath)下面应包含解压出来的easypruner文件夹。例如路径为 /home/zhaoxu/pruning-tools/easypruner, 则
打开bashrc:
vim ~/.bashrc
添加路径:
# 工具包路径在:/home/zhaoxu/pruning-tools
export PYTHONPATH=/home/zhaoxu/pruning-tools/easypruner:$PYTHONPATH
使上述修改生效:
source ~/.bashrc