update req & setup.py

Signed-off-by: robbie228 <v_lmli@webank.com>

python/requirements-fate.txt

@@ -1,23 +1,26 @@
-lmdb==1.3.0
-torch==1.13.1
+--extra-index-url https://download.pytorch.org/whl/cpu
+
+lmdb==1.4.1
+torch==2.3.1+cpu
 fate_utils
 pydantic==1.10.12
 cloudpickle==2.1.0
 click
 ruamel-yaml==0.16
-numpy
+numpy==1.26.4
 pandas==2.0.3
 transformers==4.37.2
 accelerate==0.27.2
 beautifultable
 requests<2.26.0
-scikit-learn
+scikit-learn==1.4.1.post1
 omegaconf
-rich
+rich==13.7.1
 opentelemetry-api
 opentelemetry-sdk
 opentelemetry-exporter-otlp-proto-grpc
 mmh3==3.0.0
 protobuf==4.24.4
 grpcio==1.59.3
-safetensors==0.4.1
+safetensors==0.4.1
+scipy

python/setup.py

@@ -33,15 +33,17 @@
 ]
 extras_require = {
     "rabbitmq": ["pika==1.2.1"],
-    "pulsar": ["pulsar-client==2.10.2"],
+    "pulsar": ["pulsar-client==3.4.0"],
     "spark": ["pyspark"],
     "eggroll": [
-        "cloudpickle",
+        "cloudpickle==2.1.0",
         "lmdb",
-        "protobuf",
-        "grpcio",
-        "grpcio-tools",
-        "protobuf",
+        "protobuf==4.24.4",
+        "grpcio==1.62.1",
+        "grpcio-tools==1.62.1",
+        "protobuf==4.24.4",
+        "psutil>=5.7.0",
+        "pynvml==11.5.0"
     ],
     "fate_flow": ["fate_flow[rabbitmq,pulsar,spark,eggroll]"],
 }