off-policy icon indicating copy to clipboard operation
off-policy copied to clipboard

RuntimeError: CUDA error: an illegal memory access was encountered

Open b762927 opened this issue 2 years ago • 0 comments

Traceback (most recent call last): File "train_mpe.py", line 157, in main(sys.argv[1:]) File "train_mpe.py", line 147, in main total_num_steps = runner.run() File "D:\off-policy-release\offpolicy\runner\mlp\base_runner.py", line 153, in run env_info = self.collecter(explore=True, training_episode=True, warmup=False) File "D:\off-policy-release\offpolicy\runner\mlp\mpe_runner.py", line 145, in shared_collect_rollout self.train() File "D:\off-policy-release\offpolicy\runner\mlp\base_runner.py", line 189, in batch_train train_info, new_priorities, idxes = update(p_id, sample) File "D:\off-policy-release\offpolicy\algorithms\maddpg\maddpg.py", line 117, in shared_train_policy_on_batch rewards = to_torch(rewards).to(**self.tpdv).view(-1, 1) RuntimeError: CUDA error: an illegal memory access was encountered When I run the maddpg, it encount cuda error

b762927 avatar Jul 08 '22 07:07 b762927