deddoggo commited on
Commit
8f9f687
·
1 Parent(s): 601879f
Files changed (1) hide show
  1. environment.yml +2 -1
environment.yml CHANGED
@@ -1,6 +1,7 @@
1
  # environment.yml
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
@@ -8,7 +9,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
 
1
  # environment.yml
2
  name: myapp-env
3
  channels:
4
+ - nvidia # Channel chính cho cudatoolkit
5
  - pytorch # Kênh chính cho faiss-gpu và pytorch (nếu cài qua conda)
6
  - conda-forge # Cho các gói khác và có thể là fallback
7
  - defaults
 
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