diff --git a/src/sagemaker/chainer/README.rst b/src/sagemaker/chainer/README.rst index 248c4f82ff..8778850209 100644 --- a/src/sagemaker/chainer/README.rst +++ b/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 | +-----------------------------+-------------+ diff --git a/src/sagemaker/mxnet/README.rst b/src/sagemaker/mxnet/README.rst index 70434aadaa..2f0dfa487a 100644 --- a/src/sagemaker/mxnet/README.rst +++ b/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 | +-------------------------+--------------+-------------+-------------+-------------+ diff --git a/src/sagemaker/pytorch/README.rst b/src/sagemaker/pytorch/README.rst index 54e8bcd510..9f908a4d37 100644 --- a/src/sagemaker/pytorch/README.rst +++ b/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 | diff --git a/src/sagemaker/tensorflow/README.rst b/src/sagemaker/tensorflow/README.rst index b56f7a989e..10b8846d98 100644 --- a/src/sagemaker/tensorflow/README.rst +++ b/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