That error sounds like there is no gpus available.
Have you noticed you wrongly wrote CUDA_VISIBLE_DEIVCES instead of CUDA_VISIBLE_DEVICES ?
If it’s a transcription error can u try torch.cuda.is_available() and torch.cuda.device_count() to check if you have available gpus/cuda when you are running that?
You should also to consider you have to allocate (as i said) both, model and input, in gpu by calling .cuda() method.