diff --git a/setup.py b/setup.py index 3632d1a..1fa327f 100644 --- a/setup.py +++ b/setup.py @@ -11,7 +11,8 @@ install_requires=[ 'numpy', 'matplotlib', - 'opencv-python' + 'opencv-python', + 'cupy' # Spams PyPI install is currently broken # 'spams'