Issue
Need help! Trying to run a python script with a pre-trained model (ESRGAN), but every time I face this error message. What could be the problem?
Before that I've installed the following dependencies:
- pip install pytorch cuda
- pip install opencv-python glob2
Solution
Fixed the problem by tweaking a code a bit, changing a device type from 'cuda' to 'cpu'.
device = torch.device('cpu')
Answered By - warch1LD
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.