一、问题描述
使用Pycharm在跑一个深度学习网络时,出现以下报错:
Process finished with exit code -1073740791 (0xC0000409)
二、问题分析
出现以上报错,一般显卡显存不足导致的。所以可以通过以下几步来逐一排查:
- 判断自己的环境是否配置好,特别是cuda、cudnn
- 判断显卡驱动是否安装好或者需要更新
- 如果是windows下,可以打开任务管理器,查看GPU显存。
如下图所示,可以发现我的电脑有两个GPU,其中 GPU 0 是集成显卡, GPU 1 是独立显卡。一般而言,我们都采用性能更加强悍的独立显卡来训练网络。所以在主程序中,第一次训练时,最好要指定显卡;如果不指定的话,可能系统默认使用GPU 0,因此导致显存不足。
三、解决方案
指定使用 GPU 1 进行训练, 通需要在主程序前加入添加以下代码:文章来源:https://www.toymoban.com/news/detail-641802.html
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
当然,如果你有多张显卡,你也可以一起进行指定:文章来源地址https://www.toymoban.com/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张显卡处理
到了这里,关于Process finished with exit code -1073740791 (0xC0000409)报错解决方案的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!