使用pytorch多重处理,manager.list收到无效的设备指针错误

时间:2019-02-13 01:05:44

标签: multiprocessing pytorch reinforcement-learning

我使用pytorch多处理模块生成体验并将其存储在manager.list()对象中。该过程停止后,我尝试索引存储在全局共享列表对象中的体验,并得到以下错误消息。我可以打印全局对象,但不能为其编制索引:

Unserializable message: Traceback (most recent call last):
  File "/home/hanzheng/miniconda3/envs/erl/lib/python3.5/multiprocessing/managers.py", line 276, in serve_client
    send(msg)
  File "/home/hanzheng/miniconda3/envs/erl/lib/python3.5/multiprocessing/connection.py", line 206, in send
    self._send_bytes(ForkingPickler.dumps(obj))
  File "/home/hanzheng/miniconda3/envs/erl/lib/python3.5/multiprocessing/reduction.py", line 50, in dumps
    cls(buf, protocol).dump(obj)
  File "/home/hanzheng/miniconda3/envs/erl/lib/python3.5/site-packages/torch/multiprocessing/reductions.py", line 213, in reduce_tensor
    (device, handle, storage_size_bytes, storage_offset_bytes) = storage._share_cuda_()
RuntimeError: invalid device pointer: 0x204aa0000 at /opt/conda/conda-bld/pytorch_1544173631724/work/aten/src/THC/THCCachingAllocator.cpp:301

有解决方案吗?谢谢。

0 个答案:

没有答案