search

snap-stanford / neural-subgraph-learning-GNN

329 stars 64 forks source link

windows multiprocessing error #9

Open lihuiliullh opened 3 years ago

lihuiliullh commented 3 years ago

I run this code in windows and get a multiprocessing error.

RuntimeError: cuda runtime error (801) : operation not supported at C:\w\1\s\windows\pytorch\torch/csrc/generic/StorageSharing.cpp:245

Do you have windows version code?

qema commented 3 years ago

Unfortunately the code is not directly supported in windows. PyTorch itself appears to have some difficulties on windows -- it may be possible to get it to run by setting num_workers=0 for all DataLoaders and using the command line --n_workers=1 for all scripts, but I'm unable to test this. Alternatively, you could try setting --device=cpu to avoid cuda altogether, or running the code through windows subsystem for linux? Let me know if these still don't work.