This PR adds the torchaudio PyTorch extension library to the torch container build process.
The other main PyTorch extension library, torchvision, was already included, but its build has been updated as well to include proper PNG/JPEG file support.
Since the nccl-tests base containers now include cuDNN, torch:nccl builds now explicitly compile PyTorch and its extensions with cuDNN support enabled.
A configurable compiler cache was also added in the build stage, to make the builds take slightly less of forever.
A 15 GiB cache cuts around an hour off of build times in my testing. PyTorch recommends 25 GiB.
It isn't known how well the GitHub Actions runner will handle it (if at all), so it's being added here starting at 1 GiB.
This is adjustable with a docker build argument, so it can be increased or decreased fairly easily later on.
New Libraries
torchaudio
PyTorch extension library to thetorch
container build process.torchvision
, was already included, but its build has been updated as well to include proper PNG/JPEG file support.nccl-tests
base containers now include cuDNN,torch:nccl
builds now explicitly compile PyTorch and its extensions with cuDNN support enabled.docker build
argument, so it can be increased or decreased fairly easily later on.