when i print torch.cuda.device_count() it shows 4. But when inside a process i do torch.cuda.set_device(1)(it works fine in a single process code), it gives multiprocesssing error(just tells the file name in traceback, nothing else)
"
Process Process-2:
Traceback (most recent call last):
File “/tools/anaconda3/envs/py35/lib/python3.5/multiprocessing/process.py”, line 252, in
_bootstrap
"
torch.cuda.set_device(0) works absolutely fine though.
Could you please provide any help in this regard?