Using the Deep Learning Base AMI - Deep Learning AMI

Using the Deep Learning Base AMI

Using 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 the NVIDIA CUDA 10.0 environment. You can also switch between different versions of CUDA. Refer to the following instructions for how to do this.

Configuring CUDA Versions

You can select and verify a particular CUDA version with the following bash commands.

Tip

You can verify the CUDA version by running NVIDIA's nvcc program.

$ nvcc --version
  • CUDA 11.0:

    $ sudo rm /usr/local/cuda $ sudo ln -s /usr/local/cuda-11.0 /usr/local/cuda
  • CUDA 10.2:

    $ sudo rm /usr/local/cuda $ sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda
  • CUDA 10.1:

    $ sudo rm /usr/local/cuda $ sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
  • CUDA 10.0:

    $ sudo rm /usr/local/cuda $ sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda