update

preprocess/openpose/annotator/openpose/body.py

@@ -16,6 +16,7 @@ from torchvision import transforms
 from . import util
 from .model import bodypose_model
 
+import devicetorch
 
 class Body(object):
     def __init__(self, model_path):
@@ -48,8 +49,9 @@ class Body(object):
             im = np.ascontiguousarray(im)
 
             data = torch.from_numpy(im).float()
-            if torch.cuda.is_available():
-                data = data.cuda()
+            data = devicetorch.to(torch, data)
+#            if torch.cuda.is_available():
+#                data = data.cuda()
             # data = data.permute([2, 0, 1]).unsqueeze(0).float()
             with torch.no_grad():
                 Mconv7_stage6_L1, Mconv7_stage6_L2 = self.model(data)