轉載自 https://blog.csdn.net/wanglitao588/article/details/77033659
今天在jupyter notebook上跑一個tensorflow 的CNN程序時,出現了一個錯誤,記錄如下:
錯誤:InternalError: Dst tensor is not initialized.
分析:出現這個錯誤一般是GPU內存耗盡,掛在后台進程中的IPython內核會這樣做
解決辦法:運行程序之前,先運行export CUDA_VISIBLE_DEVICES=1,僅顯卡設備1GPU可見,tensorflow 算是一個比較貪心的工具了就算用device_id指定gpu 也會占用別的GPU的顯存資源 必須在執行程序前執行
export CUDA_VISIBLE_DEVICES=n(n為可見的服務器編號)再去執行 Python 代碼.py 才不會占用別的GPU資源
export CUDA_VISIBLE_DEVICES=n(n為可見的服務器編號)再去執行 Python 代碼.py 才不會占用別的GPU資源
