update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -44,7 +44,7 @@ RUN echo "--- KIỂM TRA MÔI TRƯỜNG SAU KHI CÀI ĐẶT ---" && \
|
|
44 |
echo "Đường dẫn Python:" && which python && \
|
45 |
echo "Phiên bản Python:" && python --version && \
|
46 |
echo "--- Thông tin PyTorch: ---" && \
|
47 |
-
python -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'CUDA available for PyTorch: {torch.cuda.is_available()}'); print(f'PyTorch built with CUDA version: {torch.version.cuda if torch.cuda.is_available() else
|
48 |
echo "--- Thông tin Faiss: ---" && \
|
49 |
python -c "import faiss; print(f'Faiss version: {faiss.__version__}'); print(f'Number of GPUs available to Faiss: {faiss.get_num_gpus()}')" && \
|
50 |
echo "--- Danh sách một số gói pip quan trọng: ---" && \
|
|
|
44 |
echo "Đường dẫn Python:" && which python && \
|
45 |
echo "Phiên bản Python:" && python --version && \
|
46 |
echo "--- Thông tin PyTorch: ---" && \
|
47 |
+
python -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'CUDA available for PyTorch: {torch.cuda.is_available()}'); print(f'PyTorch built with CUDA version: {torch.version.cuda if torch.cuda.is_available() else \"N/A\"}'); print(f'cuDNN version: {torch.backends.cudnn.version() if torch.cuda.is_available() and torch.backends.cudnn.is_available() else \"N/A\"}'); print(f'Number of GPUs available to PyTorch: {torch.cuda.device_count()}')" && \
|
48 |
echo "--- Thông tin Faiss: ---" && \
|
49 |
python -c "import faiss; print(f'Faiss version: {faiss.__version__}'); print(f'Number of GPUs available to Faiss: {faiss.get_num_gpus()}')" && \
|
50 |
echo "--- Danh sách một số gói pip quan trọng: ---" && \
|