search

ufoym / deepo

Setup and customize deep learning environment in seconds.
http://ufoym.com/deepo
MIT License
6.32k stars 750 forks source link

cannot import name '_update_worker_pids' #104

Closed Kangsan-Jeon closed 5 years ago

Kangsan-Jeon commented 5 years ago

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?

ufoym commented 5 years ago

See here.

Kangsan-Jeon commented 5 years ago

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?

ufoym commented 5 years ago

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 commented 5 years ago

I really appreciate your help.