Shmuma / ptan

PyTorch Agent Net: reinforcement learning toolkit for pytorch
MIT License
530 stars 163 forks source link

DQN Speedup Windows CUDA Error #14

Open michael20at opened 6 years ago

michael20at commented 6 years ago

First, thanks for the great work! I tried the DQN Speedup files and was able to get 01 and 02 to run (with about 50fps on an GTX 1070), but anthing >=3 gives a Pytorch 32 Error for me as soon as I use Cuda, because apparently Pytorch on Windows can't use multiprocessing! (CUDA IPC not supported) Is there any way around this, because the speedup would be really great for experiments, and multiprocessing seems to be the most important part!