这个问题是在作者使用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。

这里的GPU顺序为0,所以在config.yml中的配置起了冲突,导致GPU无法被使用,报错。
解决办法:
重写config.yml的DEVICE设置,根据CUDA为GPU分配的资源顺序,酌情设置。本机中的CUDA将GPU设为0,所以config.yml中的DEVICE设置为:
DEVICE: 0 # 0: GPU, 1: CPU
修复Anime-InPainting模型训练错误
本文解决了在使用Anime-InPainting模型训练过程中遇到的设备配置问题。问题源于config.yml文件中的DEVICE设置与本机CUDA环境冲突,导致GPU无法正常工作。通过调整配置文件中的DEVICE设置以匹配CUDA分配的GPU ID,成功解决了该问题。
1363





