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

使用 Deep Learning Base AMI

使用 Deep Learning Base AMI

Base AMI 附带 GPU 驱动程序基础平台,以及可用来部署您自己的自定义深度学习环境的加速库。默认情况下,该 AMI 配置有 NVidia CUDA 9 环境。不过,通过重新配置环境变量 LD_LIBRARY_PATH,您也可以切换到 CUDA 8 环境。您只需将该环境变量字符串的 CUDA 9 部分替换为 CUDA 8 相应的内容。

配置 CUDA 版本

LD_LIBRARY_PATH 字符串的 CUDA 9 部分 (默认安装)

... :/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:/lib/nccl/cuda-9 :...LD_LIBRARY_PATH 值的其余部分

替换为 CUDA 8

... :/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:/lib/nccl/cuda-8 :...LD_LIBRARY_PATH 值的其余部分

提示

有关已知问题的信息,请参阅发行说明: