-
Notifications
You must be signed in to change notification settings - Fork 1.9k
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
Comments
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. |
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. |
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. |
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. |
Thanks for looking into this @michaelgsharp To confirm: CUDA version: 10.1 |
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. |
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
The text was updated successfully, but these errors were encountered: