PaddleHub 肺炎CT影像分析项目如何使用GPU进行预测
收藏
我发现公开项目中的肺炎CT影像分析(https://aistudio.baidu.com/aistudio/projectdetail/289819?channelType=0&channel=0)在预测的时候,不能用GPU,一次要6s,好久,请问怎么才能使用GPU去跑呢
0
收藏
请登录后评论
这个链接点开以后显示项目未公开啊?
我放错链接了,应该是https://aistudio.baidu.com/aistudio/projectdetail/289819?channelType=0&channel=0
不科学呀,我是在公开项目里面复制过来的链接
预测的results = pneumonia.segmentation(data=input_dict)这行代码改成:
results = pneumonia.segmentation(data=input_dict,use_gpu=True) 试一下~