深度学习 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-9 :...LD_LIBRARY_PATH 值的其余部分

提示

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