From c2d5d7656d60cd0c5f8cf5a02a683029a586bb22 Mon Sep 17 00:00:00 2001
From: David Doukhan <david.doukhan@gmail.com>
Date: Fri, 8 Nov 2024 16:53:04 +0100
Subject: [PATCH] and cuda

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index a835ff2..cdbd619 100644
--- a/setup.py
+++ b/setup.py
@@ -113,7 +113,7 @@
     test_suite="run_test.py",
     description = DESCRIPTION,
     license = "MIT",
-    install_requires=['tensorflow', 'numpy', 'pandas', 'scikit-image', 'pyannote.core', 'matplotlib', 'Pyro4', 'pytextgrid', 'soundfile', 'onnxruntime-gpu'], #'torch'
+    install_requires=['tensorflow[and-cuda]', 'numpy', 'pandas', 'scikit-image', 'pyannote.core', 'matplotlib', 'Pyro4', 'pytextgrid', 'soundfile', 'onnxruntime-gpu'], #'torch'
  #   keywords = "example documentation tutorial",
     url = "https://github.com/ina-foss/inaSpeechSegmenter",
 #    packages=['inaSpeechSegmenter'],