File size: 1,219 Bytes
25950e2 79d0162 25950e2 79d0162 25950e2 79d0162 6b77b8d 79d0162 25950e2 79d0162 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
# environment.yml
name: myapp-env # Hoặc tên môi trường của bạn
channels:
- nvidia # Channel chính cho cudatoolkit
- pytorch # Kênh cho faiss-gpu và pytorch-cpu (làm nền)
- conda-forge # Cho các gói khác
- defaults
dependencies:
- python=3.11.12 # Hoặc phiên bản Python của bạn
# --- CUDA Toolkit và các gói Conda chính ---
- nvidia::cudatoolkit==11.8 # Yêu cầu CUDA 11.8
# Yêu cầu faiss-gpu. Conda sẽ cố gắng tìm bản build tương thích.
# Ví dụ: pytorch::faiss-gpu=1.8.0 (nếu bạn biết phiên bản cụ thể cho py311, cuda11.8)
# Hoặc conda-forge::faiss-gpu (thường ổn định)
- conda-forge::faiss-gpu
- numpy=1.26.4 # Ghim NumPy < 2.0 để tương thích với Faiss
# Các gói PyTorch CPU làm nền (sẽ được ghi đè bởi bản GPU từ pip)
- pytorch::pytorch-cpu
- pytorch::torchvision-cpu
- pytorch::torchaudio-cpu
# --- Các gói Pip ---
- pip
- pip:
- gradio
- sentence-transformers
- unsloth
- transformers
- rank_bm25
- huggingface_hub
# KHÔNG có torch, torchvision, torchaudio, numpy ở đây
# ... các gói pip khác nếu cần |