Extra install requirements. (#164)

* Testing install requirements * GPU install requirements
2020-12-09 02:22:47 -08:00
parent 0e02f394ee
commit 34b38dd4d0
1 changed files with 8 additions and 0 deletions
@@ -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)