deddoggo commited on
Commit
b6dac79
·
1 Parent(s): 98653b4
Files changed (1) hide show
  1. environment.yml +2 -2
environment.yml CHANGED
@@ -9,10 +9,10 @@ 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
- - nvidia::cudatoolkit==11.8
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=1.9.0
16
  - numpy=1.26.4 # Hoặc numpy=1.26.4 (để tương thích với faiss cũ hơn nếu cần)
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
 
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
+ - nvidia::cudatoolkit==11.8.0
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=1.26.4 # Hoặc numpy=1.26.4 (để tương thích với faiss cũ hơn nếu cần)
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