diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_conv_pass.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_conv_pass.py
index a934c264e473f802cd3bd699b57352e7e73f3207..3a16f4f7d55bf5ed1d6ac283cf0ee3f64432f590 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_conv_pass.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_conv_pass.py
@@ -22,6 +22,8 @@ import paddle.fluid.core as core
 from paddle.fluid.core import PassVersionChecker
 from paddle.fluid.core import AnalysisConfig
 
+os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
+
 
 class TensorRTSubgraphPassConvTest(InferencePassTest):
 
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_deformable_conv.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_deformable_conv.py
index 3bed89e74f595919173a18511f7c24ca3eb217d9..6a0e98539a3c2aad8bbd05b1f483ba6132bbd7f9 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_deformable_conv.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_deformable_conv.py
@@ -14,6 +14,7 @@
 
 from __future__ import print_function
 
+import os
 import unittest
 import numpy as np
 from inference_pass_test import InferencePassTest
@@ -22,6 +23,8 @@ import paddle.fluid.core as core
 from paddle.fluid.core import PassVersionChecker
 from paddle.fluid.core import AnalysisConfig
 
+os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
+
 
 class TRTDeformableConvTest(InferencePassTest):