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
Dj3w ago
I'll look into this for you, it shouldn't be hard for us to support.
トトロ1号
トトロ1号OP3w ago
Thanks! Hope can use it soon. any update?
Dj
Dj3w ago
Yes actually, im working on this tomorrow :) I started tonight
Dj
Dj3w 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
Dj3w ago
I found a bug in our CI and I'm using this PR to fix it.
トトロ1号
トトロ1号OP3w ago
good news, thanks! I saw that the PR is still under review....
Dj
Dj3w ago
I'll ask someone to handle it in the morning :)
トトロ1号
トトロ1号OP2w ago
Is it ok to use?
Dj
Dj2w 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号OP2w ago
ok, thanks for that!

Did you find this page helpful?