GPU is not working for prediction

[tchen0623]

Hi, I meet a problem when I was predicting using the E2FGVI-HQ. My CPU and Memory working for whole time but GPU does not work at all. I have ensure my CUDA is installed successfully, and the device for this code is return cuda. device = torch.device("cuda" if torch.cuda.is_available() else "cpu")