That actually sounds like a docker issue. When PyTorch runs, we check with the CUDA driver API if the driver is sufficient, and we report back what it says…
Are you using NVIDIA-docker? because you need that to run docker images with GPU support.