diff --git a/setup.py b/setup.py
index c9928df..dc5d496 100644
--- a/setup.py
+++ b/setup.py
@@ -71,7 +71,6 @@ def get_compiler():
 ]
 
 try:
-    import torch
     from torch.utils.cpp_extension import CUDAExtension
 
     ext_modules.append(