Open melaniihariono opened 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
Try to put "--gpus 0,1,2,3" in the command
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.