|
发表于 2022-6-13 21:16:10
|
显示全部楼层
请问下载的最新版本运行GPEN出现这个错误怎么解决?
F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\cuda\__init__.py:52: UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu. (Triggered internally at ..\c10\cuda\CUDAFunctions.cpp:115.)
No CUDA runtime is found, using CUDA_HOME='F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\CUDA\'
return torch._C._cuda_getDeviceCount() > 0
Traceback (most recent call last):
File "run.py", line 137, in <module>
processer = FaceEnhancement(in_size=args.in_size, model=args.model, use_sr=args.use_sr, sr_model=args.sr_model, channel_multiplier=args.channel_multiplier, narrow=args.narrow, key=args.key, device='cuda' if args.use_cuda else 'cpu')
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\GPEN\face_enhancement.py", line 17, in __init__
self.facedetector = RetinaFaceDetection(base_dir, device)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\GPEN\face_detect\retinaface_detection.py", line 28, in __init__
self.net = self.net.to(device)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 852, in to
return self._apply(convert)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 530, in _apply
module._apply(fn)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 530, in _apply
module._apply(fn)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 552, in _apply
param_applied = fn(param)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 850, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File "F:\WeGameApps\tools\GPEN_GFPGAN_Windows\_internal\Python37\lib\site-packages\torch\cuda\__init__.py", line 172, in _lazy_init
torch._C._cuda_init()
RuntimeError: CUDA driver initialization failed, you might not have a CUDA gpu. |
|