這是錯誤的我:“RuntimeError:無法初始化CUDA分叉的子流程。使用CUDA多處理,您必須使用“衍生”開始的方法”。我用13.0 nc12s_v3集群。
我用這個:“
”,但仍然得到相同的問題。有解決方案嗎?
謝謝
嗨@phdykd,
謝謝你發布你的問題在磚社區。
一種方法是包括start_method =“叉”參數的生成函數調用如下:mp。產卵(* prev_args start_method =“叉”)。盡管這將工作,它可能提高一個警告建議使用方法(選項2)。
另一個建議的解決方案,根據PyTorch(鏈接),是使用torch.multiprocessing。start_processes: torch.multiprocessing。start_processes (* prev_args start_method =“叉”)。
請參考這個文檔為更多的細節