dgl-ke
dgl-ke copied to clipboard
RuntimeError: Cannot re-initialize CUDA in forked subprocess
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
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.