search

IDEA-Research / DWPose

"Effective Whole-body Pose Estimation with Two-stages Distillation" (ICCV 2023, CV4Metaverse Workshop)
Apache License 2.0
2.05k stars 137 forks source link

using cpu #72

Closed azamet90 closed 5 months ago

azamet90 commented 5 months ago

is there a way to make it use gpu instead of cpu?

yzd-v commented 5 months ago

Here

azamet90 commented 5 months ago

Here

thanks for reply! but im amd user and have rocm installed. that one says cuda there?

azamet90 commented 5 months ago

resim