PyTorch出现：RuntimeError: An attempt has been made to start a new process...报错

XU_MAN_2024-03-27 20:33

1.查看显卡：

第一步：同时按下键盘的win+r键，输入cmd

第二步：随后输入

python 复制代码

nvidia-smi

2.查看下载安装的包：

conda list

python 复制代码

conda list

3.问题报错

PyTorch出现：RuntimeError: An attempt has been made to start a new process...报错

RuntimeError:

An attempt has been made to start a new process before the

current process has finished its bootstrapping phase.

This probably means that you are not using fork to start your

child processes and you have forgotten to use the proper idiom

in the main module:

if name == 'main':

freeze_support()

...

The "freeze_support()" line can be omitted if the program

is not going to be frozen to produce an executable.

出错原因：网上查找结果，原因是多进程的原因

解决方法：

代码在运行 epoch 之前，加上 if __name__=='__main__'

2，不使用多线程，即去掉 num_workers 参数，或设置 num_workers=0。

python 复制代码

dataset = torch.utils.data.DataLoader(trainset, batch_size=opt.batch_size,
                                        shuffle=True, num_workers=0, drop_last=True)