Create requirements-gpu.txt
Browse filesAdded libraries required for inference on GPU.
- requirements-gpu.txt +7 -0
requirements-gpu.txt
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
pandas==2.2.2
|
| 2 |
+
keras_nlp==0.12.1
|
| 3 |
+
keras>=3.0
|
| 4 |
+
tensorflow[and-cuda]>=2.16.0
|
| 5 |
+
pandarallel==1.6.5
|
| 6 |
+
scikit-learn==1.5.0
|
| 7 |
+
nltk==3.8.1
|