ValueError: _set_worker_pids should be called only once for each _BaseDataLoaderIter.

[choieq]

My environment is pytorch 1.6, cuda 10.2 in ubuntu

I replace _update_worker_pids with _set_worker_pids in code/dataloader.py, then i got this error.
"ValueError: _set_worker_pids should be called only once for each _BaseDataLoaderIter. "

Help me ..

[Akshaypatil7]

Use pytorch=0.4.1, cuda=9.2 in ubuntu
use this command to install both:
conda install pytorch=0.4.1 cuda92 -c pytorch