search

xingyizhou / pytorch-pose-hg-3d

PyTorch implementation for 3D human pose estimation
GNU General Public License v3.0
613 stars 143 forks source link

demo error #31

Closed tsly123 closed 5 years ago

tsly123 commented 6 years ago

Hi, Thank you for your work. I am facing a error while running the demo as shown below. In addition, it took around 5mins to produce the error output.

image

I have install all the required packages.

xingyizhou commented 6 years ago

Hi, You can try python src/demo.py -demo images/h36m_1214.png .

tsly123 commented 6 years ago

Thank @xingyizhou It's working now but the processing time for my 1080 is around 5 mins/image. During the process, I use nvidia-smi to check whether the GPU work or not. The code does not use GPU resource. Is it an error or just something wrong with my machine?

xingyizhou commented 5 years ago

Hi, Sorry for the late reply. The code should run in GPU if set up correctly. 5 mins look un-reasonable even in CPU. Could you debug which line the code is stuck on? If it is model = torch.load('hgreg-3d.pth').cuda(), the problem might be the cudnn version doesn't fit your GPU, you should change to anther pytorch version (https://pytorch.org/previous-versions/).