adaptive_rag / requirements_gpu.txt
lanny xu
add react
fe717e9
# GPU优化版依赖文件 - 针对RTX 4090 Linux环境
# 安装命令: pip install -r requirements_gpu.txt
# 核心框架
langchain>=0.1.0
langgraph>=0.0.40
langchain-community>=0.0.20
langchain-core>=0.1.0
# LLM集成
langchain-ollama>=0.1.0
# 向量数据库和嵌入(GPU优化版本)
chromadb>=0.4.0
sentence-transformers>=2.2.0
# PyTorch GPU版本(CUDA 11.8支持)
--index-url https://download.pytorch.org/whl/cu118
torch>=2.0.0+cu118
torchvision>=0.15.0+cu118
torchaudio>=2.0.0+cu118
# Transformers和加速库
transformers>=4.30.0
accelerate>=0.20.0
# GPU加速向量搜索
faiss-gpu>=1.7.4
# CUDA支持库
cupy-cuda12x>=12.0.0
# 文档处理
tiktoken>=0.5.0
beautifulsoup4>=4.12.0
requests>=2.31.0
rank-bm25>=0.2.2 # BM25检索,用于混合搜索
# 网络搜索
tavily-python>=0.3.0
# 数据处理(兼容版本)
numpy>=1.24.0,<2.0
pandas>=2.0.0
# 工具库
python-dotenv>=1.0.0
pydantic>=2.0.0
typing-extensions>=4.0.0
# 系统监控
gputil>=1.4.0
psutil>=5.9.0
# 可选:Jupyter支持
jupyter>=1.0.0
ipykernel>=6.0.0
fastapi>=0.104.0
uvicorn>=0.23.0
python-multipart>=0.0.6
pyngrok>=7.0.0
cloudflared