「天工开物」AI 工具链课程（二）- 安装与环境搭建

引言

我们在上一节中已经简单介绍了地平线天工开物（Horizon OpenExplorer）AI工具链，相信您对它已经有了初步的认识。为了方便学习后面课程，使用天工开物工具链进行实战开发，还需要做一些准备工作，本节课将介绍这一部分内容：环境搭建

环境搭建

为了简化环境搭建步骤，方便大家使用天工开物（Horizon OpenExplorer）AI工具链，我们提供了docker 镜像内含算法工具包、编译器工具包及嵌入式工具包等。离线 docker镜像可以在收到的软件包 ./Docker/tensorflow-horizon-docker-${version}.tar.gz 中获取。（下载地址，关于Docker 请参考: Why Docker ? ）

使用该镜像环境，我们需要一台带有NVIDIA GPU显卡的物理机，我们建议它的操作系统为 Ubuntu(>=14.04) 或 Centos(>6.5)，此外我们需要安装以下软件:

Nvidia driver >= 410.93
Docker >= 1.1.2 (建议 18.03.0-ce)
Nvidia Docker 2.0.3

安装 Nvidia 驱动

参考如何安装Nvidia驱动程序进行安装
测试Nvida驱动

nvidia-smi -a

如果得到类似下图的结果，说明驱动已经安装成功了

安装 Docker

参考 Docker install 进行安装
测试 docker

docker run helloword

安装 Nvidia-docker

参考 Nvidia-docer install 进行安装
测试nvidia-docker

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

OpenExplorer 镜像导入

导入Horizon OpenExplorer Development Kit镜像

export version=... # 写成实际的版本号
gunzip -c tensorflow-horizon-docker-${version}.tar.gz | docker load

测试镜像

docker run hub.hobot.cc/aitools/tf-toolkit:v3.5.1.0 tree -d -L 3 /root

至此，环境安装完成。