diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
index ed939e934a91..35cda1a85543 100644
--- a/.github/workflows/integration-tests.yml
+++ b/.github/workflows/integration-tests.yml
@@ -76,10 +76,12 @@ jobs:
       - name: Install Triton
         run: |
           cd python
+          pip3 install cmake==3.24
           TRITON_USE_ASSERT_ENABLED_LLVM=TRUE pip3 install -e '.[tests]'
 
       - name: Run lit tests
         run: |
+          pip3 install lit
           cd python
           LIT_TEST_DIR="build/$(ls build)/test"
           if [ ! -d "$LIT_TEST_DIR" ]; then
diff --git a/python/setup.py b/python/setup.py
index ea0140bff9b7..1d5eb89c591d 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -211,6 +211,7 @@ def build_extension(self, ext):
 
 download_and_copy_ptxas()
 
+
 setup(
     name="triton",
     version="2.1.0",
@@ -218,12 +219,17 @@ def build_extension(self, ext):
     author_email="phil@openai.com",
     description="A language and compiler for custom Deep Learning operations",
     long_description="",
-    packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/impl", "triton/ops", "triton/runtime", "triton/ops/blocksparse"],
+    packages=[
+        "triton",
+        "triton/_C",
+        "triton/language",
+        "triton/tools",
+        "triton/impl",
+        "triton/ops",
+        "triton/runtime",
+        "triton/ops/blocksparse"],
     install_requires=[
-        "cmake>=3.20",
         "filelock",
-        "torch",
-        "lit",
     ],
     package_data={"triton": ["third_party/**/*"]},
     include_package_data=True,
@@ -242,6 +248,10 @@ def build_extension(self, ext):
     ],
     test_suite="tests",
     extras_require={
+        "build": [
+            "cmake>=3.20",
+            "lit",
+        ],
         "tests": [
             "autopep8",
             "flake8",
@@ -249,6 +259,7 @@ def build_extension(self, ext):
             "numpy",
             "pytest",
             "scipy>=1.7.1",
+            "torch",
         ],
         "tutorials": [
             "matplotlib",