search

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:
MIT License
7.27k stars 1.93k forks source link

GPU Problem #992

Open melaniihariono opened 2 years ago

melaniihariono commented 2 years ago

Hello, does anyone know? why the model train using CPU instead of GPU? I already have 4 GPU, but none of them are used for training.

Dipankar1997161 commented 2 years ago

maybe first check if cuda toolkit or cudnn is installed in your environment. That could the reason why your code does not detect any gpu. One of the best practices is to use CUDA_Available_devices to see if gpus are detected or not

YuyangLeng commented 2 years ago

Try to put "--gpus 0,1,2,3" in the command