HigherHRNet-Human-Pose-Estimation icon indicating copy to clipboard operation
HigherHRNet-Human-Pose-Estimation copied to clipboard

Published 20 hours ago • verified publisher icon HRNet

RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl

Open TIMO-AI opened this issue 4 years ago • 4 comments

When i set _C.MULTIPROCESSING_DISTRIBUTED = False, this mistake happens during train. RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu device type at start of device string: 0

TIMO-AI avatar Jun 16 '20 07:06 TIMO-AI

Same issue, any suggestion?

superying avatar Aug 14 '20 07:08 superying

Same issue, any suggestion?

enduringstack avatar Aug 20 '20 07:08 enduringstack

Is there a solution?Thank you so much if u can help me.

Cheng-11 avatar Nov 20 '20 04:11 Cheng-11

you can comment the code elif args.gpu is not None: torch.cuda.set_device(args.gpu) model = model.cuda(args.gpu) in the dist_train.py

mobeixiaoxin avatar Nov 27 '20 13:11 mobeixiaoxin