isunnyrock commited on
Commit
219b4d3
1 Parent(s): a1755dc

some requirements updates

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,9 +1,12 @@
1
  matplotlib==3.8.4
2
  seaborn==0.13.2
3
  opencv-python-headless
4
- tensorflow-gpu==2.15.0
5
  keras==2.15.0
6
  openai==1.23.2
7
  numpy==1.25.2
8
  scipy==1.11.4
9
  tensorflow-hub==0.16.1
 
 
 
 
1
  matplotlib==3.8.4
2
  seaborn==0.13.2
3
  opencv-python-headless
4
+ tensorflow[and-cuda]==2.15.0
5
  keras==2.15.0
6
  openai==1.23.2
7
  numpy==1.25.2
8
  scipy==1.11.4
9
  tensorflow-hub==0.16.1
10
+ nvidia-cudnn-cu12==8.9.4.25
11
+ CUDA-Toolkit==12.0
12
+ tensorRT==8.6.1