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.

Sign up for GitHub

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

Jump to bottom

upgrade cuda version to support tf 2.6.0 #766

Closed
davesteps opened this issue Oct 14, 2021 · 0 comments · Fixed by #913
Closed

upgrade cuda version to support tf 2.6.0 #766

davesteps opened this issue Oct 14, 2021 · 0 comments · Fixed by #913
Assignees
@0x2b3bfa0
Labels
cml-image Subcommand

Comments

@davesteps
Copy link

Hi, We were unable to use tf 2.6.0 with the dvc2-base1-gpu docker image. The problem is probably due to the image using cuda version 11.0.3 which is not compatible with tf 2.6.0 (requires 11.2). Are there plans to upgrade the cuda version?

Original question on discord for context.

Many thanks.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@0x2b3bfa0 0x2b3bfa0

Labels
cml-image Subcommand
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Bump CUDA to 11.2.1 on container images iterative/cml
3 participants
@DavidGOrtega @davesteps @0x2b3bfa0