That explains the large file sizes for the CUDA versions. I can’t believe that I hadn’t realized this before! So the only software requirement is that the installed graphics driver supports the CUDA version that you are requesting, right?
I remember that when cloning apex, the CUDA version number had to be identical (major and minor) to the version number PyTorch was built with. So in that scenario, I guess apex used the globally installed CUDA libs, and threw a warning when that differs from the PT ones. What I do not understand, though, is why apex could then not built itself using the CUDA libs that are shipped with PyTorch? Seems that I am misunderstanding something here. Doesn’t PyTorch’s included CUDA include build tools? (I am not very familiar with all subcomponents of CUDA)