GPU is not working for prediction
tchen0623 opened this issue · 0 comments
tchen0623 commented
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")