Update requirements.txt to clarify optional machine learning dependencies and provide installation notes for CPU and GPU versions of torch.
This commit is contained in:
+6
-5
@@ -49,11 +49,12 @@ plotly>=5.17.0
|
|||||||
matplotlib==3.9.0
|
matplotlib==3.9.0
|
||||||
wordcloud==1.9.3
|
wordcloud==1.9.3
|
||||||
|
|
||||||
# ===== 机器学习(可选,用于情感分析) =====
|
# ===== 机器学习(可选,用于情感分析,不安装也没事写了容错程序) =====
|
||||||
# torch>=2.0.0 # 需要单独安装CUDA版本
|
torch>=2.0.0 # CPU版本
|
||||||
# transformers>=4.30.0
|
transformers>=4.30.0
|
||||||
# scikit-learn>=1.3.0
|
scikit-learn>=1.3.0
|
||||||
# xgboost>=2.0.0
|
xgboost>=2.0.0
|
||||||
|
# NOTE:如果要安装GPU版本的torch,指令为pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126
|
||||||
|
|
||||||
# ===== 工具库 =====
|
# ===== 工具库 =====
|
||||||
python-dotenv>=1.0.0
|
python-dotenv>=1.0.0
|
||||||
|
|||||||
Reference in New Issue
Block a user