![](/img/trans.png)
[英]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.