1 | initial version |
This means that there is no CUDA library installed in the Docker container, which is required for GPU acceleration. As a result, the torch.cuda.is_available function returns False, indicating that PyTorch is unable to detect the presence of a GPU. To use a GPU, you need to install the CUDA library in the Docker container.