deddoggo commited on
Commit
7c0ee71
·
1 Parent(s): 653d0d1
Files changed (2) hide show
  1. Dockerfile +6 -7
  2. 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 bằng pip index URL cụ thể
33
- # QUAN TRỌNG: XÁC MINH LẠI URL HẬU TỐ CUDA ( dụ: `cu121` thay vì `cu126` nếu đó là chuẩn).
34
- # URL bạn cung cấp: --index-url https://download.pytorch.org/whl/cu126
35
- # Lệnh này sẽ tìm các wheel phù hợp với Python và kiến trúc trên index đó.
36
- RUN echo "Bắt đầu cài đặt PyTorch, torchvision, torchaudio với GPU (CUDA 12.1)..." && \
37
- pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 && \
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 # Quan trọng cho cudatoolkit
6
  - conda-forge
7
  - defaults
8
  dependencies:
9
- - python=3.11.12 # Hoặc phiên bản Python của bạn
10
  # --- Các gói Conda ---
11
- # Yêu cầu faiss với hỗ trợ GPU cho Linux từ conda-forge
12
- # Conda sẽ cố gắng tìm bản build faiss-gpu hoặc faiss kéo theo cudatoolkit
13
- # phiên bản CUDA tương thích với PyTorch bạn sẽ cài.
14
- - nvidia::cudatoolkit=12.6 # Phiên bản CUDA tương thích với PyTorch bạn sẽ cài
15
- - conda-forge::faiss # Đây cách gọi phổ biến cho Linux
16
- - numpy=1.26.4 # Đảm bảo numpy < 2.0 để tương thích với các gói khác
17
- # Hoặc bạn có thể thử chỉ 'conda-forge::faiss' và xem nó có tự động kéo cudatoolkit không.
18
- # Để chắc chắn, bạn có thể chỉ định cudatoolkit:
19
- # - conda-forge::faiss-gpu cudatoolkit=12.1 # Thay 12.1 bằng phiên bản CUDA PyTorch của bạn sẽ dùng
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
- # KHÔNG torch đây, sẽ cài trong Dockerfile
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
14
+ # Bạn 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, 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