From 34b38dd4d086f37dc7fc770cfd0f46210da962eb Mon Sep 17 00:00:00 2001
From: Liam <3579535@myuwc.ac.za>
Date: Wed, 9 Dec 2020 11:22:47 +0100
Subject: [PATCH] Extra install requirements. (#164)

* Testing install requirements

* GPU install requirements
---
 setup.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/setup.py b/setup.py
index 803f193044..af12147276 100644
--- 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)