这个问题是在作者使用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
这里的GPU顺序为0,所以在config.yml中的配置起了冲突,导致GPU无法被使用,报错。

解决办法:

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

相关文章:

  • 2021-07-17
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-04-14
  • 2022-12-23
  • 2021-08-27
猜你喜欢
  • 2021-05-09
  • 2022-12-23
  • 2022-12-23
  • 2022-01-08
  • 2022-12-23
  • 2021-06-03
  • 2022-12-23
相关资源
相似解决方案