GPU Problem
melaniihariono opened this issue · comments
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.
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