1. 程式人生 > >tensorflow:錯誤 InternalError: Dst tensor is not initialized

tensorflow:錯誤 InternalError: Dst tensor is not initialized

電腦環境

GPU:GTX1060/6G

系統:ubuntu14.04

python2.7

tensorflow1.4.0

cuda8.0

cudnn6.0

 

在跑https://github.com/MarvinTeichmann/KittiSeg.git的demo時

執行python demo.py --input_image data/demo/demo.png後,報錯

Dst tensor is not initialized

分析:可能是我同時執行的其他程式也用到了GPU,造成GPU資源不夠或衝突了。

可以以限制使用GPU資源,

指定使用部分

CUDA_VISIBLE_DEVICES=1 python main.py

限制視訊記憶體使用

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))  

更詳細設定可以參考

http://www.cnblogs.com/darkknightzh/p/6591923.html

http://stackoverflow.com/questions/36668467/change-default-gpu-in-tensorflow

http://stackoverflow.com/questions/37893755/tensorflow-set-cuda-visible-devices-within-jupyter