使用深度学习基础 AMI - 深度学习 AMI
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 Amazon Web Services 服务入门

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

使用深度学习基础 AMI

使用深度学习基础 AMI

Base AMI 附带 GPU 驱动程序基础平台,以及可用来部署您自己的自定义深度学习环境的加速库。默认情况下,该 AMI 配置为采用 NVIDIA CUDA 10.0 环境。还可以在不同版本的 CUDA 之间切换。有关如何执行此操作,请参阅以下说明。

配置 CUDA 版本

您可以使用以下 bash 命令选择并验证特定 CUDA 版本。

提示

您可以通过运行 NVIDIA 的nvcc计划。

$ nvcc --version
  • 库达 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