update
Browse files- environment.yml +3 -4
environment.yml
CHANGED
@@ -2,18 +2,17 @@
|
|
2 |
name: myapp-env
|
3 |
channels:
|
4 |
- pytorch # Kênh chính cho faiss-gpu và pytorch (nếu cài qua conda)
|
5 |
-
- nvidia # Cho cudatoolkit và các phụ thuộc CUDA khác
|
6 |
- conda-forge # Cho các gói khác và có thể là fallback
|
7 |
- defaults
|
8 |
dependencies:
|
9 |
- python=3.11.12
|
10 |
# --- CUDA Toolkit và các gói Conda chính ---
|
11 |
# Yêu cầu cudatoolkit 12.1 từ channel nvidia
|
12 |
-
-
|
13 |
# Cài faiss-gpu từ channel pytorch, nó nên tương thích với CUDA 12.1
|
14 |
# Bỏ phiên bản cụ thể để lấy bản stable mới nhất tương thích
|
15 |
-
- faiss
|
16 |
-
- numpy
|
17 |
# Với Faiss mới hơn, có thể không cần ghim chặt numpy<2.0 nữa,
|
18 |
# nhưng để an toàn, bạn có thể giữ lại hoặc thử bỏ đi nếu faiss mới hỗ trợ numpy 2.x
|
19 |
# Các gói PyTorch CPU làm nền (vẫn giữ để Conda biết về sự tồn tại của PyTorch)
|
|
|
2 |
name: myapp-env
|
3 |
channels:
|
4 |
- pytorch # Kênh chính cho faiss-gpu và pytorch (nếu cài qua conda)
|
|
|
5 |
- conda-forge # Cho các gói khác và có thể là fallback
|
6 |
- defaults
|
7 |
dependencies:
|
8 |
- python=3.11.12
|
9 |
# --- CUDA Toolkit và các gói Conda chính ---
|
10 |
# Yêu cầu cudatoolkit 12.1 từ channel nvidia
|
11 |
+
- conda-forge::cudatoolkit==11.8
|
12 |
# Cài faiss-gpu từ channel pytorch, nó nên tương thích với CUDA 12.1
|
13 |
# Bỏ phiên bản cụ thể để lấy bản stable mới nhất tương thích
|
14 |
+
- faiss=1.9.0
|
15 |
+
- numpy=1.26.4 # Hoặc numpy=1.26.4 (để tương thích với faiss cũ hơn nếu cần)
|
16 |
# Với Faiss mới hơn, có thể không cần ghim chặt numpy<2.0 nữa,
|
17 |
# nhưng để an toàn, bạn có thể giữ lại hoặc thử bỏ đi nếu faiss mới hỗ trợ numpy 2.x
|
18 |
# Các gói PyTorch CPU làm nền (vẫn giữ để Conda biết về sự tồn tại của PyTorch)
|