search

xinntao / ESRGAN

ECCV18 Workshops - Enhanced SRGAN. Champion PIRM Challenge on Perceptual Super-Resolution. The training codes are in BasicSR.
https://github.com/xinntao/BasicSR
Apache License 2.0
5.95k stars 1.06k forks source link

CUDA version #26

Open achadr opened 5 years ago

achadr commented 5 years ago

I can not train the code ( ESRGAN model), and I get this error training_problem_after_fixing_dataset_problem I think it's related to CUDA version, I am using CUDA 10.0

xinntao commented 5 years ago

It seems that your cuda is not available. You can run this in python:

import torch
print(torch.cuda.is_available())

If it is not True, you may need to check your installation.