1. 程式人生 > >Tensorflow和MNIST識別自己手寫的數字 finished with exit code -1073740791 (0xC0000409)解決方案

Tensorflow和MNIST識別自己手寫的數字 finished with exit code -1073740791 (0xC0000409)解決方案

在跑alexnet的測試demo時,遇到無錯跳出finished with exit code -1073740791 (0xC0000409),查出是顯示卡記憶體不足的問題,解決如下:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

    1
    2

參考連結:
(1)https://stackoverflow.com/questions/44552585/prevent-tensorflow-from-accessing-the-gpu

(2)https://blog.csdn.net/dlhlSC/article/details/77100435