原因:
config.yml中的 DEVICE配置出了問題。
原文件中默認:DEVICE: 1 # 0: CPU, 1: GPU
這里的配置與本機CUDA的起了沖突,在CUDA安裝目錄下的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.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
這樣就可以執行GPU了