Official Runpod PyTorch 2.5 Template #78
Description
Currently there is no official Pytorch template for version 2.5+ and requires manual torch update or using an uncached container image. Pytorch 2.5 introduced some good features (for e.g. FlexAttention) that have made its way into downstream libraries like Huggingface.
4d05770 added Pytorch 2.5 as a target in the docker bake. Is it possible to push a container image for this tag? It will be quite helpful. Thanks