| # 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 |