RuntimeError: CUDA error: an illegal memory access was encountered #286
Description
Hello,
When running some of the test code, I met the following error:
% python test/discrete/test_sac.py (git)-[master]-
tianshou/data/buffer.py:348: UserWarning: ListReplayBuffer will be replaced in version 0.4.0.
warnings.warn("ListReplayBuffer will be replaced in version 0.4.0.")
Epoch #1: 0%| | 0/1000 [00:00<?, ?it/s]
Traceback (most recent call last):
File "test/discrete/test_sac.py", line 121, in <module>
test_discrete_sac()
File "test/discrete/test_sac.py", line 104, in test_discrete_sac
result = offpolicy_trainer(
File "tianshou/trainer/offpolicy.py", line 92, in offpolicy_trainer
result = train_collector.collect(n_step=collect_per_step)
File "tianshou/data/collector.py", line 270, in collect
self.data.act = to_numpy(result.act)
File "tianshou/data/utils/converter.py", line 18, in to_numpy
return x.detach().cpu().numpy()
RuntimeError: CUDA error: an illegal memory access was encounteredThe same error shows for test/discrete/test_ppo.py. However, it works well for some other RL algorithms, such as DQN (test/discrete/test_dqn.py).
Here is my environment:
- tianshou: 0.3.1 (commit f0129f4)
- pytorch: v1.6.0
- python: v3.8.5
- OS: Ubuntu 16.04