I am training a network and I put my network parameters and data loader on Cuda but when I run training, nvidia-smi shows 0% GPU usage with memory occupied. All the computation is running on CPU as CPU usage is very high. I checked every variable and networks parameters with “variable_name.is_cuda” and  “next(net.parameters()).is_cuda”  respectively in my training loop. They are all true. Can anyone help me regarding this issue?