diff --git a/setup.py b/setup.py
index bb6d104..85f97b6 100644
--- a/setup.py
+++ b/setup.py
@@ -183,5 +183,6 @@ setup(
     install_requires=[
         "torch",
         "einops",
+        "packaging",
     ],
 )