在使用PyTorch框架搭建完网络,训练时出现CUDA error (3): initialization error (multiprocessing)错误,此类错误(若确定网络搭建没问题的话)多半是因为有其它并行程序正在运行,占用,导致在运行PyTorch程序时无法启动多线程,导致报错。
本人遇到此错误是因为在运行Matlab程序(parfor循环),启动多线程,Matlab程序运行完毕后而没有手动关闭多线程,手动关闭多线程后还有残留线程在运行,最后直接退出Matlab,再运行PyTorch程序,就不再报错了。
若是不知道什么程序开启了并行进程,导致无法再开启新的多线程,则直接重启机器即可解决上述问题。
如果您觉得我的文章对您有所帮助,欢迎扫码进行赞赏!