Open avi-pomicell opened 1 year ago
Hi
I'm getting this error when trying to use dgl-ke with GPU: RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method
RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method
I am able to run the code on CPU only. Any ideas?
More info: I set up the environment using these insturctions: https://github.com/awslabs/dgl-ke/issues/267#issuecomment-1347933918
Same issue here.
XinliYu
commented
11 months ago
Hi
I'm getting this error when trying to use dgl-ke with GPU:
RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start methodI am able to run the code on CPU only. Any ideas?
More info: I set up the environment using these insturctions: https://github.com/awslabs/dgl-ke/issues/267#issuecomment-1347933918