fix tensorflow version

2024-05-06 18:17:15 +01:00
parent 7d346ba2ce
commit edd1e4c123
3 changed files with 44 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,5 @@
+tensorflow[and-cuda] == 2.15.1
+pandas
+# Make sure to install the nvidia pyindex first
+# nvidia-pyindex
+nvidia-cudnn