deddoggo commited on
Commit
601879f
·
1 Parent(s): a7d5f55
Files changed (1) hide show
  1. 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
- - nvidia::cudatoolkit
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-gpu
16
- - numpy<2.0 # 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
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)