转换TensorFlow 的PB 模型到 UFF格式
1,使用aws dlami 启动一个实例
2,source activate tensorflow_p36
3,下载TensorRT 的 tar 安装文件,参考 https://docs.nvidia.com/deeplearning/sdk/tensorrt-archived/tensorrt-515/tensorrt-install-guide/index.html#installing-tar
4,$ tar xzvf TensorRT-5.1.x.x.<os>.<arch>-gnu.cuda-x.x.cudnn7.x.tar.gz
5,$ ls TensorRT-5.1.x.x
6,export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<eg:TensorRT-5.1.x.x/lib>
7,pip install tensorrt-5.1.x.x-cp3x-none-linux_x86_64.whl
8,cd TensorRT-5.1.x.x/uff
9,pip install uff-0.6.3-py2.py3-none-any.whl
10,which convert-to-uff
11,cd TensorRT-5.1.x.x/graphsurgeon
12, pip install graphsurgeon-0.4.1-py2.py3-none-any.whl
13,convert-to-uff input_file