欢迎使用 TensorFlow 剪枝工具包(TensorFlow Pruning Toolkit),这是一个用于模型剪枝的开源工具。以下是安装步骤:

通过 pip 安装 📦

pip install tensorflow-pruning-toolkit
pip_install

从源码安装 💻

  1. 克隆仓库:
    git clone https://github.com/tensorflow/pruning-toolkit.git
    
  2. 进入目录并安装依赖:
    cd pruning-toolkit
    pip install -r requirements.txt
    
github_clone

验证安装 ✅

运行以下命令确认是否安装成功:

python -c "import pruning_toolkit; print(pruning_toolkit.__version__)"

依赖项要求 📋

  • Python 3.7+
  • TensorFlow 2.x
  • pip >= 20.0
  • 其他第三方库(详见 requirements.txt

如需进一步了解使用方法,请访问 /community/open_source/tensorflow_pruning_toolkit/usage 查看完整文档。

installation_flow