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

[CI,DOCKER] Bump gpu image to cuda 11.0.3 #8119

Merged
merged 1 commit into from
May 25, 2021

Conversation

tkonolige
Copy link
Contributor

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM as long as all test cases can pass with CUDA 11.0.
btw, are we going to maintain the compatibility to CUDA 10? If so we probably need to figure out a way to run CIs for both versions, although it seems too much for the CI. If not, then we need to make sure we list this environment change in the release note.

@tkonolige
Copy link
Contributor Author

It seems like the host version of CUDA has to match what is running in the containers (supposedly this is not the case, but making the versions match fixed all the issues we were seeing). If we want to test against CUDA 10 and 11, were going to have to partition the GPU nodes.

@jroesch jroesch merged commit b64466a into apache:main May 25, 2021
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Jun 17, 2021
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Jun 17, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants