-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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 GPU image for CUDA 11.7 #14363
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 |
LOG(WARNING) << "Unable to detect CUDA version, default to \"-mcpu=sm_20\" instead"; | ||
arch = 20; | ||
LOG(WARNING) << "Unable to detect CUDA version, default to \"-mcpu=sm_50\" instead"; | ||
arch = 50; |
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.
SM20 is no longer supported by CUDA 11.7, it's too old (more than 10 years ago gpu)
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.
This is great! Thanks for keeping cuda up-to-date!
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