[英]How to run pretrained tensorflow model using Nvidia's Tensor RT on Jetson TX1?
[英]killed error in tensorflow when I try load convolutional pretrained model in jetson tx1
我有一个在inception_resnet模型上训练过的人脸识别模型。
当我运行tensorflow代码以在Nvidia Jetson TX1上加载经过训练的模型时,它只会输出“ killed”。 我该如何调试?
我能做什么? 我认为是因为内存问题!
根据这个问题, “杰森”被“杀死”意味着它内存不足。 在TX1上可能无法运行inception_resnet模型。
您可以尝试将batch_size的数字减小,例如从32减少到16,这将减少内存消耗并增加训练时间。
终于找到答案了!
如果未设置GPU内存的最大部分,它将分配几乎全部可用内存。 我的问题是GPU内存不足。
您可以通过会话配置。
我在tf.GPUOptions中将per_process_gpu_memory_fraction配置设置为0.8,问题已解决。
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.8)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.