-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Add supported CUDA version to the documentation. #372
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
Conversation
Codecov Report
@@ Coverage Diff @@
## master #372 +/- ##
=======================================
Coverage 93.07% 93.07%
=======================================
Files 51 51
Lines 3568 3568
=======================================
Hits 3321 3321
Misses 247 247 Continue to review full report at Codecov.
|
src/sagemaker/tensorflow/README.rst
Outdated
@@ -860,7 +860,7 @@ The TensorFlow Docker images support Python 2.7. They include the following Pyth | |||
- tensorflow | |||
- tensorflow-serving-api | |||
|
|||
The Docker images extend Ubuntu 16.04. | |||
The Docker images extend Ubuntu 16.04. GPU image uses CUDA 9.0. |
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.
"The CPU Docker images extend the Ubuntu 16.04 Docker image. The GPU images extend the CUDA 9.0 Docker image."
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.
Both CPU and GPU extend ubuntu 16.04 image :)
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.
then just add CUDA to the list above and change "Python packages" to "dependencies". Or add a table like the other READMEs have
* Add supported CUDA version to the documentation about ML frameworks containers. * Move tensorflow cuda dependency.
Add supported CUDA version to the documentation about ML frameworks containers.
Merge Checklist
Put an
x
in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.