环境配置

需要基于PyTorch1.1以上的版本,并安装onnx库:

  1. 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