default
Browse files- Dockerfile +0 -7
Dockerfile
CHANGED
|
@@ -15,13 +15,6 @@ RUN conda env create -f environment.yml && \
|
|
| 15 |
# Bước 5: Kích hoạt môi trường Conda
|
| 16 |
SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"]
|
| 17 |
|
| 18 |
-
# Bước 6: Cài đặt/nâng cấp PyTorch, torchvision, torchaudio với GPU (CUDA 11.8)
|
| 19 |
-
# PHIÊN BẢN PYTORCH CẦN ĐƯỢC XÁC MINH TẠI https://download.pytorch.org/whl/cu118
|
| 20 |
-
RUN echo "Bắt đầu cài đặt/nâng cấp PyTorch, torchvision, torchaudio với GPU (CUDA 11.8)..." && \
|
| 21 |
-
pip install --no-cache-dir --upgrade \
|
| 22 |
-
torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 \
|
| 23 |
-
--extra-index-url https://download.pytorch.org/whl/cu118 && \
|
| 24 |
-
echo "Đã cài đặt/nâng cấp PyTorch, torchvision, torchaudio cho cu118."
|
| 25 |
|
| 26 |
# Bước 7: Xác thực cài đặt (RẤT QUAN TRỌNG)
|
| 27 |
# Output ở đây sẽ cho bạn biết PyTorch có nhận CUDA 11.8 không.
|
|
|
|
| 15 |
# Bước 5: Kích hoạt môi trường Conda
|
| 16 |
SHELL ["conda", "run", "-n", "myapp-env", "/bin/bash", "-c"]
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
# Bước 7: Xác thực cài đặt (RẤT QUAN TRỌNG)
|
| 20 |
# Output ở đây sẽ cho bạn biết PyTorch có nhận CUDA 11.8 không.
|