Open 1-NoLimits opened 2 years ago
Hi,
I set device to cuda and default also to cuda but it runs on cpu. Further inspection shows .cpu() hardcoded across the scrips.
Is there a version that uses cuda?
If you set the device parameter equal to "cuda:0" or simply move the model and and data to the GPU (e.g. model = model.cuda(); data = data.cuda()) this runs fine on GPU as cuda-enabled.
Hi,
I set device to cuda and default also to cuda but it runs on cpu. Further inspection shows .cpu() hardcoded across the scrips.
Is there a version that uses cuda?