-
Notifications
You must be signed in to change notification settings - Fork 3.7k
[CI] Update CUDA to 11.7 #14293
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
[CI] Update CUDA to 11.7 #14293
Conversation
|
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment. Generated by tvm-bot |
|
My experience with CUDA and the whole CUDA ecosystem is very limited, but just wanted to point out that it would be good to review also the tlcpack CUDA versions, perhaps to propose this new one and deprecate some of the old version. Can you have a look? |
|
Do you mean this https://github.com/tlc-pack/tlcpack? I'm not familiar with this package, but its README shows how to build an image with a specific CUDA version. Is there other action required to support newer CUDA versions in tlcpack? |
driazati
left a comment
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.
the tlc-pack/tlcpack repo is separate and doesn't currently have a good strategy to verify changes, so I don't think we should block this PR on that. The updates there on the Docker images used to build packages can happen as a follow up
|
@driazati It seems the driver hasn't been updated, can you help? |
|
The GPU nodes should be using the new image now that https://github.com/tlc-pack/ci/actions/runs/4440725353 is done, though it may take some time (i.e. a day or so) to flush out the old, currently running nodes. If you check the |
|
It passed! |
Following the docker file update in #14293, I'm doing the actual image update. Validated in https://ci.tlcpack.ai/blue/organizations/jenkins/tvm-gpu/detail/ci-docker-staging/10/pipeline
The last time we updated the CUDA version on CI was two years ago, I think it is a good time for another update.
v11.7 is the latest version officially supported by PyTorch (https://pytorch.org/get-started/locally/). Rather than jumping directly to v12, I think it is better to stay in the stable v11 line for now.
This would unblock the PR #14291 which uses a relatively new feature in cuBLAS introduced in v11.4.