swz30 / Restormer

[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

windows上训练

ShaunLeeJ opened this issue · comments

Traceback (most recent call last):
File "basicsr/train.py", line 321, in
main()
File "basicsr/train.py", line 195, in main
prefetcher = CPUPrefetcher(train_loader)
File "d:\xlj\restormer-main\basicsr\data\prefetch_dataloader.py", line 72, in init
self.loader = iter(loader)
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch\utils\data\dataloader.py", line 355, in iter
return self._get_iterator()
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch\utils\data\dataloader.py", line 301, in _get_iterator
return _MultiProcessingDataLoaderIter(self)
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch\utils\data\dataloader.py", line 914, in init
Traceback (most recent call last):
File "", line 1, in
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\spawn.py", line 105, in spawn_main
exitcode = _main(fd)
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\spawn.py", line 114, in _main
prepare(preparation_data)
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\spawn.py", line 225, in prepare
_fixup_main_from_path(data['init_main_from_path'])
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\spawn.py", line 277, in _fixup_main_from_path
run_name="mp_main")
File "E:\Anaconda\envs\Restormer\lib\runpy.py", line 263, in run_path
pkg_name=pkg_name, script_name=fname)
File "E:\Anaconda\envs\Restormer\lib\runpy.py", line 96, in _run_module_code
mod_name, mod_spec, pkg_name, script_name)
File "E:\Anaconda\envs\Restormer\lib\runpy.py", line 85, in run_code
exec(code, run_globals)
File "D:\XLJ\Restormer-main\basicsr\train.py", line 7, in
import torch
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch_init
.py", line 123, in
raise err
OSError: [WinError 1455] ҳ▒▒▒ļ▒̫С▒▒▒޷▒▒▒ɲ▒▒▒▒▒ Error loading "E:\Anaconda\envs\Restormer\lib\site-packages\torch\lib\cusolverMg64_11.dll" or one of its dependencies.
w.start()
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\process.py", line 112, in start
self._popen = self._Popen(self)
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\context.py", line 223, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\context.py", line 322, in _Popen
return Popen(process_obj)
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\popen_spawn_win32.py", line 89, in init
reduction.dump(process_obj, to_child)
File "E:\Anaconda\envs\Restormer\lib\multiprocessing\reduction.py", line 60, in dump
ForkingPickler(file, protocol).dump(obj)
BrokenPipeError: [Errno 32] Broken pipe
Traceback (most recent call last):
File "E:\Anaconda\envs\Restormer\lib\runpy.py", line 193, in _run_module_as_main
"main", mod_spec)
File "E:\Anaconda\envs\Restormer\lib\runpy.py", line 85, in _run_code
exec(code, run_globals)
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch\distributed\launch.py", line 340, in
main()
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch\distributed\launch.py", line 326, in main
sigkill_handler(signal.SIGTERM, None) # not coming back
File "E:\Anaconda\envs\Restormer\lib\site-packages\torch\distributed\launch.py", line 301, in sigkill_handler
raise subprocess.CalledProcessError(returncode=last_return_code, cmd=cmd)
subprocess.CalledProcessError: Command '['E:\Anaconda\envs\Restormer\python.exe', '-u', 'basicsr/train.py', '--local_rank=0', '-opt', 'Motion_Deblurring/Options/Deblurring_Restormer.yml', '--launcher', 'pytorch']' returned non-zero exit status 1.
Killing subprocess 44220