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。
RuntimeError: cuda runtime error (30) : unknown errorr_第1张图片
这里的GPU顺序为0,所以在config.yml中的配置起了冲突,导致GPU无法被使用,报错。

解决办法:

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

你可能感兴趣的:(学习经验分析)