Extra install requirements. (#164)

* Testing install requirements * GPU install requirements
2026-06-12 07:55:14 +08:00 · 2020-12-09 11:22:47 +01:00
parent 0e02f394ee
commit 34b38dd4d0
1 changed files with 8 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@@ -21,4 +21,12 @@ setup(name='tinygrad',
      ],
      install_requires=['numpy', 'requests'],
      python_requires='>=3.8',
+      extras_require={
+        'gpu': ["pyopencl"],
+        'testing': [
+            "pytest",
+            "torch",
+            "tqdm",
+        ],
+      },
      include_package_data=True)