@vitis-ai docker_gpu 国内编译教程
欢迎来到恒殿您好!欢迎来到恒殿分享。
docker的安装步骤一
步骤二
如果新安装的系统或者 没有安装docker 就直接选择步骤一中的官方脚本安装方式。
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
- 成果展示成果展示
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)