{ai}[foss/2024a] PyTorch v2.6.0 w/ CUDA 12.6.0#23433
{ai}[foss/2024a] PyTorch v2.6.0 w/ CUDA 12.6.0#23433jpecar wants to merge 2 commits intoeasybuilders:developfrom
Conversation
Updated software
|
|
WIP based on #22824 with cuda bits added. No tests were run so this should be treated only as a baseline. I need it to proceed with building some other tools ... |
|
I'm working on the CUDA version which isn't as easy as this PR because e.g. Triton is required. I'm using PyTorch 2.7 currently for my tests to avoid conflicting requirements. |
Are you saying a PyTorch 2.6 with CUDA using |
Not fully sure. 2.6 and 2.7 require different Triton versions so I'm working on the latter. Can't remember anymore if it really was incompatible but wanted to focus my efforts on one version anyway. |
That means we'll end up with a situation where we have PyTorch 2.6 CPU-only and PyTorch 2.7 w/ CUDA? Focusing on PyTorch 2.7 makes more sense of course, but then we should also have that CPU-only... |
Maybe. I asked before if we should have a module titled "[software] with strong GPU acceleration" that has no GPU support and just install the CUDA version only. |
|
it's missing the info depends on: |
|
@jpecar Does it make sense to pursue this, since we have PyTorch 2.7.1 already? |
I have no hard feelings for it, if no one else needs it, I'm happy to see it dropped. |
In that case, I think we should close this (and I would prefer if you do that yourself, to make it clear you're fine with it) |
(created using
eb --new-pr)