CUDA error: out of memory about elegantrl HOT 3 CLOSED

Hi,

Running the run.py in both the main directory and in the MultiGPU directory leads me to have an error:

We haven't finished checking the multi-GPU version yet, so we put these files in elegantrl/AgentZoo/ElegantRL-MultiGPU.
Once all the checks are done, we'll update it directly to the elegantrl directory.

from elegantrl.

Hi,

Running the run.py in both the main directory and in the MultiGPU directory leads me to have an error:

We haven't finished checking the multi-GPU version yet, so we put these files in elegantrl/AgentZoo/ElegantRL-MultiGPU.
Once all the checks are done, we'll update it directly to the elegantrl directory.

I just wanted to make sure you saw, I referred to both the main directory and the multigpu directory. Out of the box, elegantrl/run.py does not work for me with the same out of memory above error. I tried both to see if one example would work.

I am unable to find a configuration that works. I have tried lowering the net size, batch size, rollout size, etc.

Traceback (most recent call last):
  File "/usr/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap
    self.run()
  File "/usr/lib/python3.8/multiprocessing/process.py", line 108, in run
    self._target(*self._args, **self._kwargs)
  File "elegantrl/run.py", line 414, in mp_train
    agent.init(net_dim, state_dim, action_dim)
  File "/home/kyle/trading/erl2/ElegantRL/elegantrl/agent.py", line 687, in init
    self.act = ActorPPO(net_dim, state_dim, action_dim).to(self.device)
  File "/home/kyle/.virtualenv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 673, in to
    return self._apply(convert)
  File "/home/kyle/.virtualenv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 387, in _apply
    module._apply(fn)
  File "/home/kyle/.virtualenv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 387, in _apply
    module._apply(fn)
  File "/home/kyle/.virtualenv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 409, in _apply
    param_applied = fn(param)
  File "/home/kyle/.virtualenv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 671, in convert
    return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
RuntimeError: CUDA error: out of memory

from elegantrl.

We have fully upgraded ElegantRL and now supports multiple GPU training (1~8 GPU).
And we have optimized the architecture of this library so that it takes up less GPU memory than before.

Now the problem you mentioned has been resolved. I'm sorry that we have been busy developing the 80 GPU version (Cloud platform) of ElegantRL, and we were unable to reply to you in time.

I will close this question in 3 days.

from elegantrl.