繁体   English   中英

当我在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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM