NVIDIA-AI-IOT / tf_to_trt_image_classification

Image classification with NVIDIA TensorRT from TensorFlow models.
BSD 3-Clause "New" or "Revised" License
455 stars 155 forks source link

models_to_frozen_graphs.py killed #42

Open map0te opened 5 years ago

map0te commented 5 years ago

Running tensorflow1.13, CUDA 10, and python3

Jetson Nano

~/Desktop/tf_to_trt_image_classification$ python3 scripts/models_to_frozen_graphs.py Converting vgg_16 {'model': <function vgg_16 at 0x7f2d7e81e0>, 'arg_scope': <function vgg_arg_scope at 0x7f78c67950>, 'num_classes': 1000, 'input_name': 'input', 'output_names': ['vgg_16/fc8/BiasAdd'], 'input_width': 224, 'input_height': 224, 'input_channels': 3, 'preprocess_fn': <function preprocess_vgg at 0x7f2d75cd90>, 'postprocess_fn': <function postprocess_vgg at 0x7f2d75ce18>, 'checkpoint_filename': 'data/checkpoints/vgg_16.ckpt', 'frozen_graph_filename': 'data/frozen_graphs/vgg_16.pb', 'trt_convert_status': 'works', 'plan_filename': 'data/plans/vgg_16.plan'} 2019-06-20 15:05:59.799985: W tensorflow/core/platform/profile_utils/cpu_utils.cc:98] Failed to find bogomips in /proc/cpuinfo; cannot determine CPU frequency 2019-06-20 15:05:59.802687: I tensorflow/compiler/xla/service/service.cc:161] XLA service 0x2c3b6ff0 executing computations on platform Host. Devices: 2019-06-20 15:05:59.802777: I tensorflow/compiler/xla/service/service.cc:168] StreamExecutor device (0): <undefined>, <undefined> 2019-06-20 15:05:59.947895: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:965] ARM64 does not support NUMA - returning NUMA node zero 2019-06-20 15:05:59.948185: I tensorflow/compiler/xla/service/service.cc:161] XLA service 0x2ad35380 executing computations on platform CUDA. Devices: 2019-06-20 15:05:59.948250: I tensorflow/compiler/xla/service/service.cc:168] StreamExecutor device (0): NVIDIA Tegra X1, Compute Capability 5.3 2019-06-20 15:05:59.949514: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1433] Found device 0 with properties: name: NVIDIA Tegra X1 major: 5 minor: 3 memoryClockRate(GHz): 0.9216 pciBusID: 0000:00:00.0 totalMemory: 3.86GiB freeMemory: 1.51GiB 2019-06-20 15:05:59.949596: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu devices: 0 2019-06-20 15:06:04.930047: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] Device interconnect StreamExecutor with strength 1 edge matrix: 2019-06-20 15:06:04.930119: I tensorflow/core/common_runtime/gpu/gpu_device.cc:990] 0 2019-06-20 15:06:04.930152: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1003] 0: N 2019-06-20 15:06:04.930426: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 697 MB memory) -> physical GPU (device: 0, name: NVIDIA Tegra X1, pci bus id: 0000:00:00.0, compute capability: 5.3) 2019-06-20 15:06:04.934580: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu devices: 0 2019-06-20 15:06:04.934683: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] Device interconnect StreamExecutor with strength 1 edge matrix: 2019-06-20 15:06:04.934720: I tensorflow/core/common_runtime/gpu/gpu_device.cc:990] 0 2019-06-20 15:06:04.934753: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1003] 0: N 2019-06-20 15:06:04.934868: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 697 MB memory) -> physical GPU (device: 0, name: NVIDIA Tegra X1, pci bus id: 0000:00:00.0, compute capability: 5.3) WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version. Instructions for updating: Colocations handled automatically by placer. WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py:1266: checkpoint_exists (from tensorflow.python.training.checkpoint_management) is deprecated and will be removed in a future version. Instructions for updating: Use standard file APIs to check for files with this prefix. Killed