tensorflow:錯誤 InternalError: Dst tensor is not initialized
阿新 • • 發佈:2018-11-12
電腦環境
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