转换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
相关推荐
Micusd 2020-11-19
xjtukuixing 2020-10-27
lybbb 2020-10-15
lybbb 2020-09-29
ghjk0 2020-09-24
yamaxifeng 2020-09-09
GDGYZL 2020-08-28
lybbb 2020-08-28
Icevivian 2020-08-25
comwayLi 2020-08-16
carbon0 2020-08-16
源式羽语 2020-08-09
sherry颖 2020-08-01
songbinxu 2020-07-19
sherry颖 2020-07-18
Niteowl 2020-07-15
Kindle君 2020-07-15
源式羽语 2020-07-04
源式羽语 2020-06-28