fix build
Browse files- Dockerfile +1 -3
- 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 --
|
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 |
-
-
|
15 |
-
-
|
|
|
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 là 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
|