Skip to content
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

Merged
merged 3 commits into from
Feb 10, 2025

Conversation

jacobtomlinson
Copy link
Member

@jacobtomlinson jacobtomlinson commented Feb 6, 2025

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

@jacobtomlinson jacobtomlinson requested a review from a team as a code owner February 6, 2025 18:43
@jacobtomlinson jacobtomlinson marked this pull request as draft February 6, 2025 18:43
@jacobtomlinson
Copy link
Member Author

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.

@jacobtomlinson
Copy link
Member Author

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.

@jacobtomlinson jacobtomlinson marked this pull request as ready for review February 6, 2025 19:11
@jacobtomlinson
Copy link
Member Author

I tried 15 again with CUDA 12 packages and things are working. I've updated the PR with these newer versions.

Copy link
Contributor

@ncclementi ncclementi left a 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.

@jacobtomlinson
Copy link
Member Author

jacobtomlinson commented Feb 10, 2025

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?

@ncclementi
Copy link
Contributor

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.

@jacobtomlinson jacobtomlinson merged commit 20478bf into rapidsai:main Feb 10, 2025
3 checks passed
@jacobtomlinson jacobtomlinson deleted the databricks-2502-tweaks branch February 10, 2025 17:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Databricks ML Runtime 15 fails to import cudf
2 participants