test

Dockerfile

@@ -3,17 +3,26 @@
 # Bước 1: Chọn base image
 FROM continuumio/miniconda3:latest
 ENV DEBIAN_FRONTEND=noninteractive
+
+# Bước 2: Thiết lập thư mục làm việc
 WORKDIR /app
+
+# Bước 3: Sao chép file định nghĩa môi trường Conda
 COPY environment.yml .
 
-# Bước 4: Tạo môi trường Conda từ file environment.yml
-# Thay "myapp-env" bằng tên môi trường bạn đã đặt trong environment.yml nếu khác.
-# Thêm --force để đảm bảo ghi đè môi trường cũ nếu có (hữu ích khi build lại).
-RUN conda env create -f environment.yml && \
+# Cài đặt mamba vào môi trường base của Conda từ conda-forge
+RUN conda install -n base -c conda-forge mamba -y
+
+# Bước 4: Tạo môi trường Conda bằng mamba
+RUN mamba env create -f environment.yml && \
     conda clean -afy
 
 # Bước 5: Kích hoạt môi trường Conda
-SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"]
+SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"] # Replace myapp-env if your env name is different
+
+# ... (Rest of your Dockerfile: pip install torch, verification steps, etc.) ...
+# Ensure your PyTorch pip install (Step 6) still targets the same CUDA version
+# (e.g., cu118 if your environment.yml has cudatoolkit=11.8.0)
 
 
 # Bước 7: Xác thực cài đặt (RẤT QUAN TRỌNG)

environment.yml

@@ -1,33 +1,26 @@
 # environment.yml
 name: myapp-env
 channels:
-  - nvidia       # Channel chính cho cudatoolkit
-  - pytorch      # Kênh chính cho faiss-gpu và pytorch (nếu cài qua conda)
-  - conda-forge  # Cho các gói khác và có thể là fallback
+  - nvidia
+  - pytorch
+  - conda-forge
   - defaults
 dependencies:
   - python=3.11.12
-  # --- CUDA Toolkit và các gói Conda chính ---
-  # Yêu cầu cudatoolkit 12.1 từ channel nvidia
-  - nvidia::cudatoolkit==11.8.0
-  # Cài faiss-gpu từ channel pytorch, nó nên tương thích với CUDA 12.1
-  # Bỏ phiên bản cụ thể để lấy bản stable mới nhất tương thích
-  - faiss
-  - numpy=1.26.4                 # Hoặc numpy=1.26.4 (để tương thích với faiss cũ hơn nếu cần)
-                               # Với Faiss mới hơn, có thể không cần ghim chặt numpy<2.0 nữa,
-                               # nhưng để an toàn, bạn có thể giữ lại hoặc thử bỏ đi nếu faiss mới hỗ trợ numpy 2.x
-  # Các gói PyTorch CPU làm nền (vẫn giữ để Conda biết về sự tồn tại của PyTorch)
-  - pytorch::pytorch-cpu
+  - nvidia::cudatoolkit=11.8.0   # Pin specific cudatoolkit
+  # Find a specific faiss-gpu version for py311 & cuda11.8
+  # e.g., pytorch::faiss-gpu=1.8.0 (verify availability)
+  - pytorch::faiss-gpu # Or a more specific version like =1.8.0 
+  - numpy=1.26.4             # Pin NumPy for faiss compatibility
+  # Pin CPU PyTorch versions if known
+  - pytorch::pytorch-cpu # >=2.0.0,<2.4.0 or specific version
   - pytorch::torchvision-cpu
   - pytorch::torchaudio-cpu
-  # --- Các gói Pip ---
   - pip
   - pip:
-    - gradio
-    - sentence-transformers
-    - unsloth
-    - transformers
-    - rank_bm25
-    - huggingface_hub
-    # KHÔNG có torch, torchvision, torchaudio, numpy ở đây
-    # ... các gói pip khác nếu cần
+    - gradio # Pin: gradio==X.Y.Z
+    - sentence-transformers # Pin
+    - unsloth # Pin
+    - transformers # Pin
+    - rank_bm25 # Pin
+    - huggingface_hub # Pin