运行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.