簡體   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