torch\csrc\cuda\Module.cpp line=34 error=30 : unknown error

3200 단어 torch

torch\csrc\cuda\Module.cpp line=34 error=30 : unknown error
 
 
THCudaCheck FAIL file=..\torch\csrc\cuda\Module.cpp line=34 error=30 : unknown error Traceback (most recent call last):   File "", line 1, in   File "D:\ProgramData\Miniconda3\lib\multiprocessing\spawn.py", line 105, in spawn_main     exitcode = _main(fd)   File "D:\ProgramData\Miniconda3\lib\multiprocessing\spawn.py", line 114, in _main     prepare(preparation_data)   File "D:\ProgramData\Miniconda3\lib\multiprocessing\spawn.py", line 225, in prepare     _fixup_main_from_path(data['init_main_from_path'])   File "D:\ProgramData\Miniconda3\lib\multiprocessing\spawn.py", line 277, in _fixup_main_from_path     run_name="__mp_main__")   File "D:\ProgramData\Miniconda3\lib\runpy.py", line 263, in run_path     pkg_name=pkg_name, script_name=fname)   File "D:\ProgramData\Miniconda3\lib\runpy.py", line 96, in _run_module_code     mod_name, mod_spec, pkg_name, script_name)   File "D:\ProgramData\Miniconda3\lib\runpy.py", line 85, in _run_code     exec(code, run_globals)   File "E:\github\Person_reID_baseline_pytorch\Person_reID_baseline_pytorch-master\test.py", line 71, in     torch.cuda.set_device(gpu_ids[0])   File "D:\ProgramData\Miniconda3\lib\site-packages\torch\cuda\__init__.py", line 264, in set_device     torch._C._cuda_setDevice(device) RuntimeError: cuda runtime error (30) : unknown error at ..\torch\csrc\cuda\Module.cpp:34 Traceback (most recent call last):   File "E:/github/Person_reID_baseline_pytorch/Person_reID_baseline_pytorch-master/test.py", line 241, in     gallery_feature = extract_feature(model,dataloaders['gallery'])   File "E:/github/Person_reID_baseline_pytorch/Person_reID_baseline_pytorch-master/test.py", line 142, in extract_feature     for data in dataloaders:   File "D:\ProgramData\Miniconda3\lib\site-packages\torch\utils\data\dataloader.py", line 819, in __iter__     return _DataLoaderIter(self)   File "D:\ProgramData\Miniconda3\lib\site-packages\torch\utils\data\dataloader.py", line 560, in __init__     w.start()   File "D:\ProgramData\Miniconda3\lib\multiprocessing\process.py", line 105, in start     self._popen = self._Popen(self)   File "D:\ProgramData\Miniconda3\lib\multiprocessing\context.py", line 223, in _Popen     return _default_context.get_context().Process._Popen(process_obj)   File "D:\ProgramData\Miniconda3\lib\multiprocessing\context.py", line 322, in _Popen     return Popen(process_obj)   File "D:\ProgramData\Miniconda3\lib\multiprocessing\popen_spawn_win32.py", line 65, in __init__     reduction.dump(process_obj, to_child)   File "D:\ProgramData\Miniconda3\lib\multiprocessing\reduction.py", line 60, in dump     ForkingPickler(file, protocol).dump(obj) BrokenPipeError: [Errno 32] Broken pipe
 
이유도num_workers=16
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize,
                                         shuffle=False, num_workers=16) for x in ['gallery','query']}

좋은 웹페이지 즐겨찾기