deddoggo commited on
Commit
d7c2175
·
1 Parent(s): 7c0ee71
Files changed (1) hide show
  1. environment.yml +2 -2
environment.yml CHANGED
@@ -1,14 +1,14 @@
1
  # environment.yml
2
  name: myapp-env
3
  channels:
4
- - pytorch
5
  - nvidia
 
6
  - conda-forge
7
  - defaults
8
  dependencies:
9
  - python=3.11.12
10
  # --- Các gói Conda ---
11
- - nvidia::cudatoolkit=12.1 # << Thay đổi thành 12.1
12
  - conda-forge::faiss-gpu # << Đổi thành faiss-gpu
13
  - numpy=1.26.4 # Giữ nguyên, Conda sẽ quản lý
14
  # Bạn có thể thêm pytorch-cpu, torchvision-cpu, torchaudio-cpu ở đây nếu muốn có base
 
1
  # environment.yml
2
  name: myapp-env
3
  channels:
 
4
  - nvidia
5
+ - pytorch
6
  - conda-forge
7
  - defaults
8
  dependencies:
9
  - python=3.11.12
10
  # --- Các gói Conda ---
11
+ - nvidia::cudatoolkit # << Thay đổi thành 12.1
12
  - conda-forge::faiss-gpu # << Đổi thành faiss-gpu
13
  - numpy=1.26.4 # Giữ nguyên, Conda sẽ quản lý
14
  # Bạn có thể thêm pytorch-cpu, torchvision-cpu, torchaudio-cpu ở đây nếu muốn có base