There is a problem with the DEVICE configuration in config.yml
.
Default in the original file: DEVICE: 1 # 0: CPU, 1: GPU
The configuration here conflicts with the native CUDA. In the CUDA installation directory, execute the `deviceQuery.exe` command in .\NVIDIA CUDA\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite
to view CUDA The ID assigned to the GPU.
The GPU order here is 0, so the configuration in config.yml conflicts, causing the GPU to be unusable and an error is reported.
Rewrite the DEVICE setting in config.yml
and set it as appropriate according to the order of resources allocated by CUDA for the GPU. CUDA in this machine sets GPU to 0, so the DEVICE in config.yml
is set to:
DEVICE: 0 # 0: GPU, 1: CPU
So that you can execute the GPU