Deep Learning AMI
Developer Guide

Using the Deep Learning Base AMI

Using to the Deep Learning Base AMI

The Base AMI comes with a foundational platform of GPU drivers and acceleration libraries to deploy your own customized deep learning environment. By default the AMI is configured with NVidia CUDA 9 environment. However, you can also switch to a CUDA 8 environment by reconfiguring the environment variable LD_LIBRARY_PATH. You simply need to replace the CUDA 9 portion of the environment variable string with its CUDA 8 equivalent.

Configuring CUDA Versions

CUDA 9 portion of the LD_LIBRARY_PATH string (installed by default)

... :/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:/lib/nccl/cuda-9 of LD_LIBRARY_PATH value

Replace with CUDA 8

... :/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:/lib/nccl/cuda-9 of LD_LIBRARY_PATH value


Refer to the release notes for information regarding known issues: