39 lines
1015 B
Bash
39 lines
1015 B
Bash
#!/bin/bash
|
|
# 安装多模态检索系统依赖包
|
|
|
|
echo "🚀 开始安装多模态检索系统依赖包..."
|
|
|
|
# 更新pip
|
|
pip install --upgrade pip
|
|
|
|
# 安装基础依赖
|
|
echo "📦 安装基础依赖包..."
|
|
pip install torch>=2.0.0 torchvision>=0.15.0
|
|
pip install transformers>=4.30.0 accelerate>=0.20.0
|
|
pip install numpy>=1.21.0 Pillow>=9.0.0
|
|
pip install scikit-learn>=1.3.0 tqdm>=4.65.0
|
|
pip install flask>=2.3.0 werkzeug>=2.3.0
|
|
pip install psutil>=5.9.0
|
|
|
|
# 安装百度VDB SDK
|
|
echo "🔗 安装百度VDB SDK..."
|
|
pip install pymochow
|
|
|
|
# 安装MongoDB驱动
|
|
echo "💾 安装MongoDB驱动..."
|
|
pip install pymongo>=4.0.0
|
|
|
|
# 安装FAISS (备用)
|
|
echo "🔍 安装FAISS..."
|
|
pip install faiss-cpu>=1.7.4
|
|
|
|
echo "✅ 依赖包安装完成！"
|
|
echo "📋 已安装的主要包："
|
|
echo " - torch (深度学习框架)"
|
|
echo " - transformers (模型库)"
|
|
echo " - pymochow (百度VDB SDK)"
|
|
echo " - flask (Web框架)"
|
|
echo " - pymongo (MongoDB驱动)"
|
|
echo ""
|
|
echo "🎯 接下来可以运行测试脚本验证安装"
|