использовать многопроцессорность pytorch, manager.list получил неверную ошибку указателя устройства - PullRequest
0 голосов
/ 13 февраля 2019

Я использую многопроцессорный модуль 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

Есть какие-нибудь решения?Спасибо.

...