Can you tell me I will use how mang GPU

uvipen / Super-mario-bros-PPO-pytorch

Proximal Policy Optimization (PPO) algorithm for Super Mario Bros

MIT License

1.07k stars 201 forks source link

Can you tell me I will use how mang GPU #13

Open 17702513221 opened 3 years ago

17702513221 commented 3 years ago

I get a error RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.95 GiB total capacity; 2.69 GiB already allocated; 10.69 MiB free; 2.78 GiB reserved in total by PyTorch)