Compiling PyTorch CUDA Extensions in Docker Without a GPU
One detail that is easy to forget when building CUDA containers is that the GPU itself is normally not available during docker build. The compiler and CUDA toolkit can be present, but PyTorch cannot inspect the target …