一、問題描述
使用Pycharm在跑一個深度學(xué)習(xí)網(wǎng)絡(luò)時,出現(xiàn)以下報錯:
Process finished with exit code -1073740791 (0xC0000409)
二、問題分析
出現(xiàn)以上報錯,一般顯卡顯存不足導(dǎo)致的。所以可以通過以下幾步來逐一排查:
- 判斷自己的環(huán)境是否配置好,特別是cuda、cudnn
- 判斷顯卡驅(qū)動是否安裝好或者需要更新
- 如果是windows下,可以打開任務(wù)管理器,查看GPU顯存。
如下圖所示,可以發(fā)現(xiàn)我的電腦有兩個GPU,其中 GPU 0 是集成顯卡, GPU 1 是獨立顯卡。一般而言,我們都采用性能更加強悍的獨立顯卡來訓(xùn)練網(wǎng)絡(luò)。所以在主程序中,第一次訓(xùn)練時,最好要指定顯卡;如果不指定的話,可能系統(tǒng)默認(rèn)使用GPU 0,因此導(dǎo)致顯存不足。
三、解決方案
指定使用 GPU 1 進行訓(xùn)練, 通需要在主程序前加入添加以下代碼:文章來源:http://www.zghlxwxcb.cn/news/detail-641802.html
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
當(dāng)然,如果你有多張顯卡,你也可以一起進行指定:文章來源地址http://www.zghlxwxcb.cn/news/detail-641802.html
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" #指定第2,3,4張顯卡處理
到了這里,關(guān)于Process finished with exit code -1073740791 (0xC0000409)報錯解決方案的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!