Paddle jetson nano 使用CPU推理,获得结果正确,使用GPU却无法得到正常结果.

goqiplq2  于 2022-10-20  发布在  其他
关注(0)|答案(9)|浏览(364)

转自: PaddlePaddle/Paddle-Lite#4680

设备:jetson nano
cuda:10.2.89
opencv 4.1.1
trt:7.1.3.0
cudnn:8.0.0.180
设置predictor代码:

def create_(model, params): 

                  config = AnalysisConfig(model, params)  # 解析模型

                  config.enable_use_gpu(memory_pool_init_size_mb=100, device_id=0)

                  config.switch_ir_optim(True)   # IR优化(默认开启)

                    # 开启TensorRT预测,精度为fp32,开启int8离线量化
                     #config.enable_tensorrt_engine(
                     #                            workspace_size = 1<<30,
                     #                            max_batch_size=1, 
                     #                            min_subgraph_size=5,
                     #                            precision_mode=AnalysisConfig.Precision.Float32,
                     #                            use_static=True, 
                     #                            use_calib_mode=False)

                 config.switch_use_feed_fetch_ops(False) # i

                 config.enable_memory_optim()

                 config.switch_specify_input_names(True)

                 predictor = create_paddle_predictor(config)

                  return predictor

将其中config.enable_use_gpu(memory_pool_init_size_mb=100, device_id=0)注解掉,可以获得正确结果.

目前在C++推理下同样出现该问题,develop分支及2.0.0*分支都已尝试。推理程序已在window(cuda10.0 cudnn7.6)下验证可得到正确结果。

qvk1mo1f

qvk1mo1f1#

您好,我们已经收到了您的问题,会安排技术人员尽快解答您的问题,请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时,您也可以通过查看 官网API文档常见问题历史IssueAI社区 来寻求解答。祝您生活愉快~

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the APIFAQGithub Issue and AI community to get the answer.Have a nice day!

hc2pp10m

hc2pp10m2#

能否提供下模型,我复现下

whhtz7ly

whhtz7ly3#

模型已发送请查收。

rkkpypqq

rkkpypqq4#

met the same problem, mark for solution..

x6yk4ghg

x6yk4ghg6#

一样遇到这种问题,CPU模式下识别正常,使用GPU识别结果都是错的,或者根本识别不到

https://github.com/PaddlePaddle/PaddleOCR/issues/2749

gxwragnw

gxwragnw7#

楼上的你们的问题解决了吗?

bakd9h0s

bakd9h0s8#

一样的问题·····不过现在GPU可以推理但是无法开启cudnn和tensor加速

cotxawn7

cotxawn79#

请问这个问题解决了吗 我也遇到一样的问题,x86下的GPU推理正常,jetson nano的GPU推理结果不正常:
PaddlePaddle/PaddleOCR#6395

相关问题