search

greatlog / SWAHR-HumanPose

Bottom-up Human Pose Estimation
127 stars 20 forks source link

cuda #12

Open yuan243212790 opened 2 years ago

yuan243212790 commented 2 years ago

RuntimeError: 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.

my torch.cuda.is_available() return true,why it can not work .thanks

DanBmh commented 2 years ago

In my case I could solve it by setting the --world_size flag to the number of gpus (defaults to 8) in the testing call: python tools/dist_valid.py --world_size 1 ...

slimxun commented 3 months ago

same problem，have you fixed it？

slimxun commented 3 months ago

In my case I could solve it by setting the --world_size flag to the number of gpus (defaults to 8) in the testing call: python tools/dist_valid.py --world_size 1 ...

thanks，it work