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

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

使用深度学习库 AMI

使用深度学习库 AMI

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

配置 CUDA 版本

你可以通过运行 NVIDIA 来验证 CUDA 版本nvcc程序。

nvcc --version

您可以使用以下 bash 命令选择和验证特定的 CUDA 版本:

sudo rm /usr/local/cuda sudo ln -s /usr/local/cuda-11.0 /usr/local/cuda

有关更多信息,请参阅 。DLAMI Base 发布说明.