深度学习 AMI
开发人员指南
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 AWS 服务入门

使用 Deep Learning Base AMI

使用 Deep Learning Base AMI

Base AMI 附带 GPU 驱动程序基础平台,以及可用来部署您自己的自定义深度学习环境的加速库。默认情况下,该 AMI 配置为采用 NVIDIA CUDA 10 环境。但是,您也可以切换到 CUDA 9 或 CUDA 8 环境。有关如何执行此操作,请参阅以下说明。

配置 CUDA 版本

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

  • CUDA 10:

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

    $ sudo rm /usr/local/cuda sudo ln -s /usr/local/cuda-9.2 /usr/local/cuda
  • CUDA 9:

    $ sudo rm /usr/local/cuda sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
  • CUDA 8:

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

提示

然后,您可以通过运行 NVIDIA 的 deviceQuery 程序验证 CUDA 版本。

$ cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make ./deviceQuery