no idea
Browse files- 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 |
- 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
|