diff --git a/setup.py b/setup.py
index 1001cee..f238566 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
 
 setup(
     name="stanford-stk",
-    version="0.0.6",
+    version="0.0.7",
     author="Trevor Gale",
     author_email="tgale@stanford.edu",
     description="Sparse Toolkit",
@@ -25,7 +25,7 @@
     ],
     packages=find_packages(),
     install_requires=[
-        "triton==2.1.0",
+        "triton>=2.1.0",
     ],
     extras_require=extra_deps,
 )