[BUG] How do I run VIBE on CPU?

mkocabas / VIBE

Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"

https://arxiv.org/abs/1912.05656

Other

2.85k stars 551 forks source link

[BUG] How do I run VIBE on CPU? #293

Closed TypeDefinition closed 4 months ago

TypeDefinition commented 4 months ago

I do not have a Nvidia GPU, and I'm constantly getting the error

Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

even though I've changed all torch.load() to use map_location='cpu'.

TypeDefinition commented 4 months ago

I stopped caring.