Add supported CUDA version to the documentation. (#372)

* Add supported CUDA version to the documentation about ML frameworks containers.

* Move tensorflow cuda dependency.

Author: nadiaya

src/sagemaker/chainer/README.rst

@@ -639,7 +639,7 @@ The Chainer Docker images have the following dependencies installed:
 +-----------------------------+-------------+
 | chainermn                   | 1.2.0       |
 +-----------------------------+-------------+
-| CUDA                        | 9.0         |
+| CUDA (GPU image only)       | 9.0         |
 +-----------------------------+-------------+
 | cupy                        | 4.0.0       |
 +-----------------------------+-------------+

src/sagemaker/mxnet/README.rst

@@ -590,7 +590,7 @@ The Docker images have the following dependencies installed:
 +-------------------------+--------------+-------------+-------------+-------------+
 | Python                  |   2.7 or 3.5 |   2.7 or 3.5|   2.7 or 3.5|   2.7 or 3.5|
 +-------------------------+--------------+-------------+-------------+-------------+
-| CUDA                    |          9.0 |         9.0 |         9.0 |         9.0 |
+| CUDA (GPU image only)   |          9.0 |         9.0 |         9.0 |         9.0 |
 +-------------------------+--------------+-------------+-------------+-------------+
 | numpy                   |       1.13.3 |      1.13.3 |      1.13.3 |      1.14.5 |
 +-------------------------+--------------+-------------+-------------+-------------+

src/sagemaker/pytorch/README.rst

@@ -667,6 +667,8 @@ The PyTorch Docker images have the following dependencies installed:
 +-----------------------------+---------------+
 | botocore                    | 1.10.35       |
 +-----------------------------+---------------+
+| CUDA (GPU image only)       | 9.0           |
++-----------------------------+---------------+
 | numpy                       | 1.14.3        |
 +-----------------------------+---------------+
 | Pillow                      | 5.1.0         |

src/sagemaker/tensorflow/README.rst

@@ -850,6 +850,7 @@ The TensorFlow Docker images support Python 2.7. They include the following Pyth
 
 - boto3
 - botocore
+- CUDA 9.0 (GPU image only)
 - grpcio
 - numpy
 - pandas