deddoggo commited on
Commit
653d0d1
·
1 Parent(s): 354901b
Files changed (2) hide show
  1. Dockerfile +1 -3
  2. environment.yml +3 -2
Dockerfile CHANGED
@@ -34,9 +34,7 @@ SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"]
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 --no-cache-dir \
38
- torch torchvision torchaudio \
39
- --extra-index-url https://download.pytorch.org/whl/cu121 && \
40
  echo "Đã cài đặt PyTorch, torchvision, torchaudio."
41
 
42
  # Bước 7: Bước xác thực cài đặt (RẤT QUAN TRỌNG để gỡ lỗi trên Spaces)
 
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)
environment.yml CHANGED
@@ -11,8 +11,9 @@ dependencies:
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
  # có phiên bản CUDA tương thích với PyTorch bạn sẽ cài.
14
- - conda-forge::faiss-gpu # Đây cách gọi phổ biến cho Linux
15
- - numpy<2.0 # Đảm bảo numpy < 2.0 để tương thích với các gói khác
 
16
  # Hoặc bạn có thể thử chỉ 'conda-forge::faiss' và xem nó có tự động kéo cudatoolkit không.
17
  # Để chắc chắn, bạn có thể chỉ định cudatoolkit:
18
  # - 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
 
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
  # có 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