diff --git a/tests/python/unittest/test_meta_schedule_tune_relay.py b/tests/python/unittest/test_meta_schedule_tune_relay.py
index 7e6f89dfb149..56b6feae067b 100644
--- a/tests/python/unittest/test_meta_schedule_tune_relay.py
+++ b/tests/python/unittest/test_meta_schedule_tune_relay.py
@@ -21,6 +21,11 @@
 import numpy as np
 from typing import Tuple, List
 
+try:
+    import torch
+except ModuleNotFoundError:
+    pass
+
 import tvm
 from tvm import relay
 from tvm.ir import IRModule