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_s
相關文章
相關標籤/搜索