search

NVlabs / PoseCNN-PyTorch

PyTorch implementation of the PoseCNN framework
Other
195 stars 49 forks source link

How to run this code on a CPU ? #2

Closed hrishikeshkale4 closed 3 years ago

hrishikeshkale4 commented 3 years ago

I don't have a GPU. So, when I run python setup.py install in lib/layers I get the following run time error -

RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50

Could you please let me know how can I run this code on a machine without a GPU ? Is it possible ?

Any help in this regard is highly appreciated !

Thanks Hrishikesh

yuxng commented 3 years ago

The code does not support running without GPUs.

hrishikeshkale4 commented 3 years ago

Thank you so much for your response. Since I don't have access to a GPU and would like to run in this on a CPU, is there a way I can integrate the C++ layers (i.e. the non-cuda layers) in your repository here (https://github.com/yuxng/PoseCNN/tree/master/lib) and integrate them with the pytorch code in this repository ?

Looking forward to hearing from you !

Thanks Hrishikesh