diff --git a/tests/test_gpu_mtp.py b/tests/test_gpu_mtp.py
index acb15e2..ab03a61 100644
--- a/tests/test_gpu_mtp.py
+++ b/tests/test_gpu_mtp.py
@@ -1,3 +1,7 @@
+import pytest
+
+pytest.importorskip("torch")
+
 import torch
 
 from gpu.mtp import MTPHead, mtp_named