The VAI pruner binaries are included in the Vitis™ AI Optimizer package. The Vitis AI Optimizer requires a commercial license to run. Contact Xilinx® FAE or sales to get a package or one separate tool for one framework.
vai_p_caffe
Install the required dependencies before running vai_p_caffe.
$ sudo apt-get install -y libopenblas-dev libgflags-dev libgoogle-glog-dev libopencv-dev libprotobuf-dev protobuf-compiler libleveldb-dev liblmdb-dev libhdf5-dev libsnappy-dev libssl-dev libboost-all-devvai_p_tensorflow
vai_p_tensorflow is developed based
on TensorFlow 1.12. Similar to tensorflow, A built vai_p_tensorflow whl package is include in VAI optimizer. Install
vai_p_tensorflow using pip install.
$ pip install tensorflow_gpu-1.12.3-cp36-cp36m-linux_x86_64.whlvai_p_darknet
vai_p_darknet binary is in our VAI
optimizer package. Once GPU environment is ready, it can be run directly.