When I ran the AlphaPose, it comes out a cuda runtime error (71). The details:
Loading YOLO model…
THCudaCheck FAIL file=…/torch/csrc/generic/StorageSharing.cpp line=232 error=71 : operation not supported
Traceback (most recent call last):
File “demo.py”, line 50, in
det_loader = DetectionLoader(data_loader, batchSize=args.detbatch).start()
File “/home/wu/AlphaPose/dataloader.py”, line 309, in start
p.start()
File “/usr/lib/python3.6/multiprocessing/process.py”, line 105, in start
self._popen = self._Popen(self)
File “/usr/lib/python3.6/multiprocessing/context.py”, line 223, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File “/usr/lib/python3.6/multiprocessing/context.py”, line 291, in _Popen
return Popen(process_obj)
File “/usr/lib/python3.6/multiprocessing/popen_forkserver.py”, line 35, in init
super().init(process_obj)
File “/usr/lib/python3.6/multiprocessing/popen_fork.py”, line 19, in init
self._launch(process_obj)
File “/usr/lib/python3.6/multiprocessing/popen_forkserver.py”, line 47, in _launch
reduction.dump(process_obj, buf)
File “/usr/lib/python3.6/multiprocessing/reduction.py”, line 60, in dump
ForkingPickler(file, protocol).dump(obj)
File “/home/wu/.local/lib/python3.6/site-packages/torch/multiprocessing/reductions.py”, line 214, in reduce_tensor
(device, handle, storage_size_bytes, storage_offset_bytes) = storage.share_cuda()
RuntimeError: cuda runtime error (71) : operation not supported at …/torch/csrc/generic/StorageSharing.cpp:232
And I have add these to ~/.bashrc :
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
Have anybody met this problem and know how to deal with it? Please give me help!!