【vitis-ai docker_gpu 国内编译教程】

恒殿

已于 2022-02-11 20:06:59 修改

阅读量2.4k

点赞数 2

文章标签：人工智能 github linux docker arm

于 2022-01-30 17:07:30 首次发布

本文链接：https://blog.csdn.net/weixin_38961173/article/details/122751089

版权

@vitis-ai docker_gpu 国内编译教程

欢迎来到恒殿

您好！欢迎来到恒殿分享。

docker的安装

步骤一
 步骤二
如果新安装的系统或者没有安装docker 就直接选择步骤一中的官方脚本安装方式。

vitis -ai 官方路径

xilinx github vitis-ai 链接

从仓库拉取项目

git clone --recurse-submodules https://github.com/Xilinx/Vitis-AI  git clone --recurse-submodules https://github.com/Xilinx/Vitis-AI

查看所有分支
```
git branch -a
```
切换分支，本教程使用的分支为1.3.2
```
git checkout -b dev remotes/origin/1.3.2
```

进入构建脚本目录

cd Vitis-AI/setup/docker/cd Vitis-AI/setup/docker/

将目录下的脚本docker_build_gpu.sh中的 docker build 后面的 --network=host 参数去掉，本人采用的是局域网内代理所以代理到dockerfile中配置。经验证这种方案是最简单有效的。
切换到存放 dockerfile的目录
```
cd docker
```

配置 docker build的代理、apt的源切换到国内，选择清华源，阿里源有小毛病下载包的过程中会出现部分包出现未知错误。将如下代码插入到目录下DockerfileGPU 文件第二行。

 			ENV MY_PROXY_URL="http://192.168.0.102:7890"
 			ENV HTTP_PROXY=$MY_PROXY_URL \
 			    HTTPS_PROXY=$MY_PROXY_URL \
 			    FTP_PROXY=$MY_PROXY_URL \
 			    http_proxy=$MY_PROXY_URL \
 			    https_proxy=$MY_PROXY_URL \
 			    ftp_proxy=$MY_PROXY_URL
 			RUN  sed -i s@/archive.ubuntu.com/@/mirrors.tuna.tsinghua.edu.cn/@g /etc/apt/sources.list
 			RUN  apt-get clean
 			RUN apt-get update

其中192.168.0.102:7890 为局域网代理服务器，有软路由科学上网的伙伴就只需要后三行就可以了

最后给像我一样电脑配置一般的伙伴准备的将DockerfileGPU文件中的make -j 修改成 make -j4 或者更小的数值，在编译protobuf的时候会爆内存，我是16G 加8700（6核12线程）。。。
现在就可以愉快的编译了。回到上一级目录，并执行编译命令
```
cd ../
./docker_build_gpu.sh
```
1. 成果展示
2. 成果展示