RuntimeError: cuda runtime error (30) : unknown errorr

这个问题是在作者使用Anime-InPainting上的模型进行训练时发现的。

原因:

config.yml中的 DEVICE配置出了问题。
原文件中默认:DEVICE: 1 # 0: CPU, 1: GPU
这里的配置与本机CUDA的起了冲突,在CUDA安装目录下的.\NVIDIA CUDA\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite 执行deviceQuery.exe指令,可以查看CUDA为GPU分配的ID。
CUDA
这里的GPU顺序为0,所以在config.yml中的配置起了冲突,导致GPU无法被使用,报错。

解决办法:

重写config.yml的DEVICE设置,根据CUDA为GPU分配的资源顺序,酌情设置。本机中的CUDA将GPU设为0,所以config.yml中的DEVICE设置为:
DEVICE: 0 # 0: GPU, 1: CPU

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值