cuda

2022-12-20

报错

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the ‘spawn’ start method

在主程序入口,加上: torch.multiprocessing.set_start_method('spawn', force=True)

这里主程序是值服务最开始执行的地方,比如 Sanic 是在 server.py 中