update
Browse files- Dockerfile +6 -7
- environment.yml +13 -16
Dockerfile
CHANGED
@@ -29,13 +29,12 @@ RUN conda env create -f environment.yml && \
|
|
29 |
# Tên môi trường "myapp-env" phải khớp với tên trong environment.yml
|
30 |
SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"]
|
31 |
|
32 |
-
# Bước 6: Cài đặt PyTorch, torchvision, torchaudio với GPU
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
echo "Đã cài đặt PyTorch, torchvision, torchaudio."
|
39 |
|
40 |
# Bước 7: Bước xác thực cài đặt (RẤT QUAN TRỌNG để gỡ lỗi trên Spaces)
|
41 |
# Bước này sẽ in ra thông tin về môi trường, giúp bạn kiểm tra xem mọi thứ có đúng không.
|
|
|
29 |
# Tên môi trường "myapp-env" phải khớp với tên trong environment.yml
|
30 |
SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"]
|
31 |
|
32 |
+
# Bước 6: Cài đặt PyTorch, torchvision, torchaudio với GPU (nhắm đến CUDA 12.1)
|
33 |
+
RUN echo "Bắt đầu cài đặt/nâng cấp PyTorch, torchvision, torchaudio với GPU (CUDA 12.1)..." && \
|
34 |
+
pip install --no-cache-dir --upgrade \
|
35 |
+
torch torchvision torchaudio \
|
36 |
+
--extra-index-url https://download.pytorch.org/whl/cu121 && \
|
37 |
+
echo "Đã cài đặt/nâng cấp PyTorch, torchvision, torchaudio."
|
|
|
38 |
|
39 |
# Bước 7: Bước xác thực cài đặt (RẤT QUAN TRỌNG để gỡ lỗi trên Spaces)
|
40 |
# Bước này sẽ in ra thông tin về môi trường, giúp bạn kiểm tra xem mọi thứ có đúng không.
|
environment.yml
CHANGED
@@ -2,31 +2,28 @@
|
|
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 |
-
#
|
12 |
-
#
|
13 |
-
#
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
#
|
18 |
-
#
|
19 |
-
#
|
20 |
-
# # (ví dụ, nếu pip install torch ... --index-url .../whl/cu121)
|
21 |
-
# ... các gói conda khác ...
|
22 |
- pip
|
23 |
- pip:
|
24 |
- gradio
|
25 |
- sentence-transformers
|
26 |
-
- numpy
|
27 |
- unsloth
|
28 |
- transformers
|
29 |
- rank_bm25
|
30 |
- huggingface_hub
|
31 |
-
#
|
32 |
-
# KHÔNG có faiss ở đây
|
|
|
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
|
15 |
+
# nhưng với pip install --upgrade trong Dockerfile, có thể không quá cần thiết.
|
16 |
+
# - pytorch::pytorch-cpu
|
17 |
+
# - pytorch::torchvision-cpu
|
18 |
+
# - pytorch::torchaudio-cpu
|
19 |
+
# --- Các gói Pip ---
|
|
|
|
|
20 |
- pip
|
21 |
- pip:
|
22 |
- gradio
|
23 |
- sentence-transformers
|
24 |
+
# - numpy # << ĐÃ XÓA
|
25 |
- unsloth
|
26 |
- transformers
|
27 |
- rank_bm25
|
28 |
- huggingface_hub
|
29 |
+
# Các gói pip khác nếu cần
|
|