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

[GPU] What is the required minimum version of CUDA? #5594

Closed
luisquintanilla opened this issue Jan 20, 2021 · 6 comments · Fixed by #5635
Closed

[GPU] What is the required minimum version of CUDA? #5594

luisquintanilla opened this issue Jan 20, 2021 · 6 comments · Fixed by #5635
Labels
documentation Related to documentation of ML.NET P3 Doc bugs, questions, minor issues, etc.

Comments

@luisquintanilla
Copy link
Contributor

Raising a question based on docs issue dotnet/docs#21251

Do the existing docs need to be updated with another CUDA version? https://github.com/dotnet/machinelearning/blob/master/docs/api-reference/tensorflow-usage.md

@michaelgsharp michaelgsharp added the question Further information is requested label Jan 21, 2021
@michaelgsharp
Copy link
Member

I am not sure. I know the version of tensorflow was updated. Let me see if I can figure out what version of CUDA tensorflow wants.

@michaelgsharp
Copy link
Member

michaelgsharp commented Jan 21, 2021

So tensorflow.net claims they want 10.0, and tensorflow itself supports lots of versions. I'll try and run/install with the version we have listed and see what happens.

@michaelgsharp
Copy link
Member

Ok, after looking into more detailed docs and testing it myself, it does require at least 10.1 for CUDA. Looking into cuDNN more now.

@michaelgsharp
Copy link
Member

So I was able to get it to work with the updated CUDA version, 10.1, but still leaving cuDNN as the original version 7.6.4.

I can't right now, but I'll get PR created that will update this documentation and the link.

@michaelgsharp michaelgsharp added documentation Related to documentation of ML.NET P3 Doc bugs, questions, minor issues, etc. and removed question Further information is requested labels Jan 21, 2021
@luisquintanilla
Copy link
Contributor Author

Thanks for looking into this @michaelgsharp

To confirm:

CUDA version: 10.1
cuDNN: 7.6.4

@michaelgsharp
Copy link
Member

That is correct. On the nvidia website they have 10.1, 10.1 update 1, and 10.1 update 2. I tested with 10.1 update 2, but in theory they should all work.

@ghost ghost locked as resolved and limited conversation to collaborators Mar 17, 2022
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
documentation Related to documentation of ML.NET P3 Doc bugs, questions, minor issues, etc.
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants