-
Notifications
You must be signed in to change notification settings - Fork 32
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update to newer Databricks ML runtime #511
Update to newer Databricks ML runtime #511
Conversation
There is also a 15 LTS runtime (and 16 LTS Beta runtime), we might want to try those out too. Marking as draft until I've tested those. IIRC the 15 Beta didn't work when I last tested it. |
I tried 15 and it failed. I'll open a new issue to track that. I think we should still merge this as is though. |
I tried 15 again with CUDA 12 packages and things are working. I've updated the PR with these newer versions. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just to double check, what didn't work before was the 15 Photon version with cuda 11?
But 15 LTS with cuda 12 is good.
Not Photon, that should always be disabled. This is the Databricks ML Runtime, which is effectively an opinionated Python environment. The v14 Runtime worked with CUDA 11, but the v15 runtime has switched to CUDA 12. Happy to merge with that clarification? |
Sounds good, I think the more info we can provide for folks to not struggle the better making choices the better. |
The documentation lists the 14.2 ML Runtime. This PR bumps to the newer 15.4 ML Runtime and updates packages to CUDA 12.
I also removed a screenshot that had the runtime version in it. We have to update this screenshot every time we use a newer version, and the screenshot didn't add a huge amount of value. So it reduces maintenance effort to just remove it.
Closes #513