fix CUDA error (3): initialization error

dgriff777 / rl_a3c_pytorch

A3C LSTM Atari with Pytorch plus A3G design

Apache License 2.0

562 stars 119 forks source link

fix CUDA error (3): initialization error #12

Closed inoryy closed 7 years ago

inoryy commented 7 years ago

I'm not sure how you've avoided the error, but without these fixes all but one workers crash with the error in title when running the example python main.py --env Pong-v0 --workers 32.

This is a fix as described in https://github.com/pytorch/pytorch/issues/2517#issuecomment-325039259

dgriff777 commented 7 years ago

This looks to be pytorch bug. torch.manual_seed shouldn't initialize cuda driver unnecessarily. If this is causing break for you just #comment the torch.manual_seed in main.py. Thanks for letting me know on this

dgriff777 commented 7 years ago

This bug is fixed in pytorch now so closing