Need Pytorch 2.5 and 2.6 offical Docker Image

We need Pytorch 2.5 and 2.6 offical Docker Image that is safe and can be used for new features such as mit-han-lab/nunchaku project.
10 Replies
Dj
Dj6mo ago
I'll look into this for you, it shouldn't be hard for us to support.
トトロ1号
トトロ1号OP6mo ago
Thanks! Hope can use it soon. any update?
Dj
Dj6mo ago
Yes actually, im working on this tomorrow :) I started tonight
Dj
Dj6mo ago
GitHub
feat: pytorch 2.6.0, version pinning by DeJayDev · Pull Request #8...
By user request, I've added a PyTorch 2.6.0 image. This PR also adds version pinning to torch dependencies for all PyTorch containers. Previously, these containers weren't actually ...
Dj
Dj6mo ago
I found a bug in our CI and I'm using this PR to fix it.
トトロ1号
トトロ1号OP6mo ago
good news, thanks! I saw that the PR is still under review....
Dj
Dj6mo ago
I'll ask someone to handle it in the morning :)
トトロ1号
トトロ1号OP6mo ago
Is it ok to use?
Dj
Dj6mo ago
@トトロ1号 Sorry for how long this is taking, I want to make our containers flow better and it's delaying me being able to release an offical RunPod 2.6.0 image. One of our employees has this template with the version you'll need built into it -> https://www.runpod.io/console/deploy?template=mm3gw8nlro&type=gpu I've alsooo sent you a DM with some credits for being patient with me while I work on our automation here, I appreciate it a lot.
トトロ1号
トトロ1号OP6mo ago
ok, thanks for that!

Did you find this page helpful?