zhang-prog 3 місяців тому
батько
коміт
f6294e2667
1 змінених файлів з 1 додано та 0 видалено
  1. 1 0
      paddlex/inference/models/common/static_infer.py

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

@@ -437,6 +437,7 @@ class PaddleInfer(StaticInfer):
                 if hasattr(config, "enable_new_ir"):
                     config.enable_new_ir(self._option.enable_new_ir)
                 config.enable_use_gpu(100, self._option.device_id)
+                config.disable_mkldnn()
                 if hasattr(config, "enable_new_executor"):
                     config.enable_new_executor()
                 # XXX: is_compiled_with_rocm() must be True on dcu platform ?