deddoggo commited on
Commit
2b9c9f4
·
1 Parent(s): 3a3ef93

update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 '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: ---" && \
 
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: ---" && \