Эх сурвалжийг харах

fix mkldnn (#4096)

Co-authored-by: bukejiyu <395822456@qq.com>
zhangyubo0722 5 сар өмнө
parent
commit
f3079d0b0b

+ 1 - 0
paddlex/inference/models/common/static_infer.py

@@ -349,6 +349,7 @@ class PaddleInfer(StaticInfer):
                     if self._option.run_mode == "paddle_fp16"
                     else PrecisionType.Float32
                 )
+                config.disable_mkldnn()
                 config.enable_use_gpu(100, self._option.device_id, precision)
                 if hasattr(config, "enable_new_ir"):
                     config.enable_new_ir(self._option.enable_new_ir)