运行GPU执行官
本文档适用于以下内容:
本文档概述了如何使用Circleci服务器运行GPU(图形处理单元)机器执行器。
先决条件
配置VM服务
在复制的管理控制台中启动启用GPU的实例。
概要
在启用NVIDIA GPU的实例上运行以下命令。以下示例使用CUDA 8.0,但您可以使用GPU实例支持的任何CUDA运行时版本。
wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-ubuntu1404-8-local_8.0.44-8-0-local_8.0.44-1_amd64-deb sudo apt-get更新导出os_release = $(uname -r)sudo apt-get安装-y linux-image-train-$ os_release linux-headers-$ os_release linux-image-$ os_release sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1__amd64-debsudo apt-get update sudo apt-get --yes - force-y是安装cuda nvidia-smi
nvidia-smi
只需要测试目的。在步骤7中安装CUDA驱动程序后,您应该很好!
将GPU步骤添加到AMI
为了避免与上述步骤相关联的启动时间,可以通过遵循以下说明来包括在AMI中配置VM服务documentation.