cannot re-initialize CUDA in forked subprocess.To use CUDA with multiprocessing,you must use the 'spawn' start method.
报错原因:
在使用sanic实现多线程操作的时候,报出这样的错误。
解决办法:
网上搜索资料发现的三种方法:
1、在main函数中设置工作模式,通过添加下面代码实现:
torch.multiprocessing.set_start_method('spawn')
2、用main包起来
import torch
def main():
for i, data in enumerate(dataloader):
# do something here
if __name__ == '__main__':
main()
3、设置num_workers为0解决
但是,上面三个方面对我的问题不起作用。
我的解决方案:
sanic版本需要设置正确,我之前用pip自动安装的版本是20.9.3,换成20.6.3版本后,问题解决。
惊不惊喜,意不意外!折磨了好多天的bug,终于解决了!!!