如何解决结合多处理运行GPU时出错
我有一个代码,它在CPU上进行了大量繁重的操作和工作,这需要使用多处理来完成。我的代码还利用了GPU(用于在Pytorch中进行预测),但出现了以下错误:
回溯(最近通话最近): 在parseResume中的文件“ /home/jumper/ParsingPlatform/ResumeParser/resume_parser.py”,行149 自我.__ get_basic_details() __get_basic_details中的第180行的文件“ /home/jumper/ParsingPlatform/ResumeParser/resume_parser.py” self.dictionaries,self .__ html_raw,self .__ sections,self .__ nlp,self .__ ner_bert 在extract_personal_details中的文件“ /home/jumper/ParsingPlatform/ResumeParser/utils.py”的第776行 cand_name,city_name,cand_email,cand_phone = ner_bert.prediction_personal(sent_mod) 文件“ /home/jumper/ParsingPlatform/ResumeParser/NER_BERT/prediction_pipeline.py”,行365,在projection_personal中 input_ids = torch.tensor([tokenized_sentence])。to(self.device) _lazy_init中的第185行的“ /opt/envParser_nlp/lib/python3.6/site-packages/torch/cuda/init.py”文件 “无法在派生子进程中重新初始化CUDA。” + msg) RuntimeError:无法在派生子进程中重新初始化CUDA。要将CUDA与多处理一起使用,必须使用“ spawn”启动方法
有人可以指导我如何从这里继续吗?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。