RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you

报错如下:
Traceback (most recent call last):
File “E2Capsnet.py”, line 777, in
main()
File “E2Capsnet.py”, line 619, in main
train(epoch)
File “E2Capsnet.py”, line 470, in train
for batch_id, (data, label) in enumerate(raf.train_loader):
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 517, in next
data = self._next_data()
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 1199, in _next_data
return self._process_data(data)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 1225, in _process_data
data.reraise()
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/_utils.py”, line 429, in reraise
raise self.exc_type(msg)
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py”, line 202, in _worker_loop
data = fetcher.fetch(index)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/utils/data/utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/utils/data/utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/media/nuaa301/YangShuang/expression_codes/Face alignment /face-alignment-master/Dataset.py”, line 38, in getitem
image_tensor = self.get_img(record)
File "/media/nuaa301/YangShuang/expression_codes/Face alignment /face-alignment-master/Dataset.py”, line 48, in get_img
feat_traindata = data_processing.feat_data(dir_img) # attention map
File “/media/nuaa301/YangShuang/expression_codes/Face alignment /face-alignment-master/data_processing.py”, line 94, in feat_data
landmarks = get_landmarks_myself(dir_img)
File “/media/nuaa301/YangShuang/expression_codes/Face alignment /face-alignment-master/data_processing.py”, line 14, in get_landmarks_myself
fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, flip_input=False)
File “/media/nuaa301/YangShuang/expression_codes/Face alignment /face-alignment-master/face_alignment/api.py”, line 76, in init
self.face_detector = face_detector_module.FaceDetector(device=device, verbose=verbose, **face_detector_kwargs)
File “/media/nuaa301/YangShuang/expression_codes/Face alignment /face-alignment-master/face_alignment/detection/sfd/sfd_detector.py”, line 31, in init
self.face_detector.to(device)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 673, in to
return self._apply(convert)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 387, in _apply
module._apply(fn)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 409, in _apply
param_applied = fn(param)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 671, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File “/home/nuaa301/anaconda3/envs/yang_torch/lib/python3.6/site-packages/torch/cuda/init.py”, line 161, in _lazy_init
"Cannot re-initialize CUDA in forked subprocess. To use CUDA with "
RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the ‘spawn’ start method

解决方案:

self.train_loader = Data.DataLoader(self.train_dataset, batch_size=batch_size, shuffle=True,   # shuffle=True之后的结果，每次都是随机打乱num_workers=0, pin_memory=True, drop_last=True)  # num_workers,使用多少个子进程来导入数据

将上述程序中的num_workers设为0即可.

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you相关推荐

Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you
RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you ...
cannot re-initialize CUDA in forked subprocess.To use CUDA with multiprocessing,you must use the ...
cannot re-initialize CUDA in forked subprocess.To use CUDA with multiprocessing,you must use the 'sp ...
RuntimeError: Cannot re-initialize CUDA in forked subprocess
问题就在于训练所使用的num_workers数量问题,直接将 num_workers=0 就不会报错了
RuntimeError: Expected object of type torch.cuda.LongTensor but found type torch.cuda.IntTensor
问题: anchor_label = np.array(self.id_dict[int(anchor_id)])identity_loss = criterion_identity(predicte ...
目的：使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU
目的:使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GPU 目的:使用CUDA环境变量CUDA_VISIBLE_DEVICES来限定CUDA程序所能使用的GP ...
ubuntu14.04.5装cuda7.5记录(解决unable to locate the kernel source，装cuda黑屏问题，装cuda循环登录问题）
ubuntu14.04.5装cuda7.5记录(解决unable to locate the kernel source,装cuda黑屏问题,装cuda循环登录问题) 参考文章: (1)ubuntu1 ...
type torch.cuda.FloatTensor but found type torch.cuda.ByteTensor
type torch.cuda.FloatTensor but found type torch.cuda.ByteTensor train_label_batch = torch.from_nump ...
pytorch CUDA driver version is insufficient for CUDA runtime version解决
pytorch CUDA driver version is insufficient for CUDA runtime version 最后发现pytorch下载时选择cuda版本,选错了安装就会报 ...
'CUDA driver version is insufficient for CUDA runtime version
ubuntu系统中,运行报错: 'CUDA driver version is insufficient for CUDA runtime version nvidia驱动是384.111, cuda ...
Check failed: error == cudaSuccess (35 vs. 0) CUDA driver version is insufficient for CUDA runtime
Check failed: error == cudaSuccess (35 vs. 0) CUDA driver version is insufficient for CUDA runtime ...

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you相关推荐

最新文章

热门文章