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.97k stars 1.06k forks source link

Error when trying to work on CUDA #98

Open Egor179 opened 4 years ago

Egor179 commented 4 years ago

When working on version 3.8.2 in CUDA mode, an error pops up, while the CPU is working fine GeForce GTX 950 E:\Games\Python38>E:\Games\Python38\test.py Model path models/RRDB_ESRGAN_x4.pth. Testing... 1 SH2DClogo Traceback (most recent call last): File "E:\Games\Python38\test.py", line 34, in output = model(imgLR).data.squeeze().float().cpu().clamp(0, 1).numpy() File "E:\Games\Python38\lib\site-packages\torch\nn\modules\module.py", line 532, in call result = self.forward(*input, kwargs) File "E:\Games\Python38\RRDBNet_arch.py", line 71, in forward trunk = self.trunk_conv(self.RRDB_trunk(fea)) File "E:\Games\Python38\lib\site-packages\torch\nn\modules\module.py", line 532, in call result = self.forward(*input, *kwargs) File "E:\Games\Python38\lib\site-packages\torch\nn\modules\container.py", line 100, in forward input = module(input) File "E:\Games\Python38\lib\site-packages\torch\nn\modules\module.py", line 532, in call result = self.forward(input, kwargs) File "E:\Games\Python38\RRDBNet_arch.py", line 47, in forward out = self.RDB1(x) File "E:\Games\Python38\lib\site-packages\torch\nn\modules\module.py", line 532, in call result = self.forward(*input, **kwargs) File "E:\Games\Python38\RRDBNet_arch.py", line 30, in forward x2 = self.lrelu(self.conv2(torch.cat((x, x1), 1))) RuntimeError: error in LoadLibraryA

He rolled back to version 3.7.5, now such an error, and the CPU is not working: E:\Games\Python>python test.py Traceback (most recent call last): File "test.py", line 5, in import torch File "E:\Games\Python\lib\site-packages\torch__init__.py", line 81, in from torch._C import * ImportError: DLL load failed: Операционная система не может запустить %1.