diff --git a/tests/test_models/test_mutables/test_mutable_channel/test_units/test_dcff_channel_unit.py b/tests/test_models/test_mutables/test_mutable_channel/test_units/test_dcff_channel_unit.py
index d0462886c..bb587c7af 100644
--- a/tests/test_models/test_mutables/test_mutable_channel/test_units/test_dcff_channel_unit.py
+++ b/tests/test_models/test_mutables/test_mutable_channel/test_units/test_dcff_channel_unit.py
@@ -9,8 +9,7 @@
 from mmrazor.structures.graph import ModuleGraph as ModuleGraph
 from .....data.models import LineModel
 
-DEVICE = torch.device('cuda:0') if torch.cuda.is_available() \
-    else torch.device('cpu')
+DEVICE = torch.device('cpu')
 
 
 class TestDCFFChannelUnit(TestCase):