diff --git a/src/gopt/test/inference.cpp b/src/gopt/test/inference.cpp
index 7551e45f..91adc3ea 100644
--- a/src/gopt/test/inference.cpp
+++ b/src/gopt/test/inference.cpp
@@ -2727,6 +2727,9 @@ TEST(TestGoptInference, ConvertFormatNCHW4) {
 
 TEST(TestGoptInference, ConvertFormatNCHW4Ic3) {
     REQUIRE_GPU(1);
+    auto cn = CompNode::load("gpu0");
+    cn.activate();
+    REQUIRE_CUDA_COMPUTE_CAPABILITY(6, 1);
     HostTensorGenerator<dtype::Float32, RandomDistribution::UNIFORM> gen{
             1.2f, 127 * 127};
     auto graph = ComputingGraph::make();
diff --git a/src/tensorrt/test/opr_replace.cpp b/src/tensorrt/test/opr_replace.cpp
index 5d03884c..fba3fbd5 100644
--- a/src/tensorrt/test/opr_replace.cpp
+++ b/src/tensorrt/test/opr_replace.cpp
@@ -1980,6 +1980,10 @@ TEST(TestTensorRTReplace, FuseConvAdd) {
 
 TEST(TestTensorRTReplace, FuseConvAddNchw2nchw4) {
     REQUIRE_GPU(1);
+    auto cn = CompNode::load("gpu0");
+    cn.activate();
+    REQUIRE_CUDA_COMPUTE_CAPABILITY(6, 1);
+
     HostTensorGenerator<dtype::Float32, RandomDistribution::UNIFORM> gen{
             1.2f, 127 * 127};
     auto graph = ComputingGraph::make();