cannot import name '_update_worker_pids'

Question

cannot import name '_update_worker_pids'

Kangsan-Jeon opened this issue 5 years ago · comments

I use your docker image(ufoym/deepo:all-jupyter) for my project.
When I test RCAN model, I receive the error message.
The message is like
---> 10 from torch._C import _set_worker_signal_handlers, _update_worker_pids,
11 _remove_worker_pids, _error_if_any_worker_fails
12 from torch.utils.data.dataloader import DataLoader

ImportError: cannot import name '_update_worker_pids'

So, I test with pytorch/pytorch docker image.
The error doesn't show again.

Is it caused by pytorch version?
If so, how can I solve the problem?

Ming commented 5 years ago

See here.

Kangsan-Jeon · Answer 1 · Thu Apr 18 2019 14:56:57 GMT+0800 (China Standard Time)

Than for your reply.
But, I don't know how to download Pytorch v1.0.0 from your image.
I just use your image by "docker pull ufoym/deepo:all-jupyter".
So, if I want to download the lower version of pytorch, what can I do?

Ming · Answer 2 · Wed Apr 24 2019 19:44:07 GMT+0800 (China Standard Time)

We only provide prebuilt images with the latest PyTorch. You can, of course, try

python generator/generate.py Dockerfile pytorch=<your version> --cuda-ver 10.0 --cudnn-ver 7

to build your own.

Kangsan-Jeon · Answer 3 · Wed Apr 24 2019 20:27:37 GMT+0800 (China Standard Time)

I really appreciate your help.