PaddlePaddle / PaddleHub

Awesome pre-trained models toolkit based on PaddlePaddle. (400+ models including Image, Text, Audio, Video and Cross-Modal with Easy Inference & Serving)【安全加固,暂停交互,请耐心等待】
https://www.paddlepaddle.org.cn/hub
Apache License 2.0
12.68k stars 2.08k forks source link

hub本地预测不能使用多卡 #1196

Open xiaomujiang opened 3 years ago

xiaomujiang commented 3 years ago

欢迎您反馈PaddleHub使用问题,非常感谢您对PaddleHub的贡献! 在留下您的问题时,辛苦您同步提供如下信息:

ubuntu18.04 python3.7.7 paddlehub 1.6.2 paddlepaddle-gpu 1.8.1.post107

使用paddlehub本地预测的时候 指定os.environ["CUDA_VISIBLE_DEVICES"] = '0,1' object_detector = hub.Module(name="yolov3_darknet53_coco2017") results = object_detector.object_detection(images=[frame_start],use_gpu=True)

但是预测过程中,都是使用的0号GPU ,没有使用另外一块, 是什么问题?

haoyuying commented 3 years ago

您好,目前hub的预测还不支持多卡操作。