一键训练 deeplabv3seg、squeezeseg、voxelnet Docker镜像安装

最新推荐文章于 2024-09-13 21:52:41 发布

华青水上

最新推荐文章于 2024-09-13 21:52:41 发布

阅读量674

点赞数

CC 4.0 BY-SA版权

分类专栏：点云深度学习文章标签：点云深度学习 Tensorflow

本文链接：https://blog.csdn.net/r1141207831/article/details/103756067

点云同时被 2 个专栏收录

8 篇文章

订阅专栏

深度学习

7 篇文章

订阅专栏

本文介绍DINK，一种集成了SOTA深度学习算法与传统算法的点云深度学习框架，支持一键运行模拟，纯Tensorflow实现，全流程模块协同与CUDA加速。详细指导了在Ubuntu16.04LTS环境下，如何安装配置NVIDIA Docker，下载并运行DINK镜像，以及启动各种深度学习任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

申明：此博客是港心时代（原太平洋AI）整理的三大点云深度学习框架，具有以下特点：

1.兼并SOTA深度学习算法与传统算法，一键运行模拟。

2.DINK中人工智能纯Tensorflow实现。

3.全流程模块协同，CUDA加速。

学习之余转载过来大家一起交流学习（如有侵权，联系删除）

建议采用

* Ubuntu16.04 LTS

* 8GB以上内存

* 至少30GB硬盘

* NVIDIA GTX GeForce GPU GTX1060TI以上

1 NVIDIA Docker安装

1.1 准备

* 安装CUDA

* 系统设置-->软件与更新-->下载选择其他-->在弹框中选择中国-->选择mirrors.aliyun.com-->右下角选择服务器

* System Settings-->Software &Updates-->Download from-->Other..-->China-mirrors.aliyun.com-->Choose Server-->Close-Reload

1.2 DOCKER CE安装

1.2.1 更新

sudo apt-get update

sudo apt-get upgrade

1.2.2 脚本安装docker

curl -fsSL get.docker.com -o get-docker.sh

sudo sh get-docker.sh --mirror Aliyun

当以下命令显示含有hello world字样时说明DOCKER环境安装

sudo docker run hello-world

1.3 下载nvidia-docker

wget http://dink.51ai.pro/nvidia-docker_1.0.1-1_amd64.deb

如果上述下载失败，下载以下

wget https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb

安装

sudo dpkg -i nvidia-docker_1.0.1-1_amd64.deb

1.4 检查nvidia-docker服务是否存在

systemctl list-units --type=service | grep -i nvidia-docker

如果上述操作失败，运行以下

systemctl list-units --type=service | grep -i nvidia-docker-plugin

安装modprobe

sudo apt-get install nvidia-modprobe

2 下载运行DINK镜像

2.1 拉取镜像前先设置镜像加速器

sudo mkdir -p /etc/docker

sudo tee /etc/docker/daemon.json <<-'EOF'
{
  "registry-mirrors": ["https://9pbu6dtx.mirror.aliyuncs.com"]
}
EOF

sudo systemctl daemon-reload

sudo systemctl restart docker

2.2 从阿里云上拉去docker镜像

sudo docker pull registry.cn-hangzhou.aliyuncs.com/dink_framework/dink0.22:latest

2.3 验证镜像：

sudo docker images

显示 registry.cn-hangzhou.aliyuncs.com/dink_framework/dink0.22 latest 即拉取成功

2.4 运行镜像并生成容器用于可视化操作：

下载运行run.sh脚本

wget http://dinkfile1.51ai.pro/run_dink6.sh

sh run_dink6.sh

2.4 在镜像中启动DINK：

进入容器后直接运行各个.sh脚本文件：

./node_deeplabv3seg_cluster.sh #deeplabv3seg_cluster一键启动节点

./node_squeezeseg_cluster.sh #squeezeseg_cluster一键启动节点

./node_voxlelnet.sh #voxlelnet一键启动节点

./dl_deeplabv3seg_train.sh #deeplabv3seg一键训练

./dl_deeplabv3seg_eval.sh #deeplabv3seg一键评估

./dl_squeezeseg_train.sh #squeezeseg一键训练

./dl_squeezeseg_eval.sh #squeezeseg一键评估

./dl_voxelnet_train.sh #voxelnet一键训练

./dl_voxelnet_eval.sh #voxelnet一键评估

./run_clion.sh #一键启动clion

./run_pycharm.sh #一键启动pycharm

(./node_euclidean_cluster.sh #euclidean cluster一键启动节点)

打开新窗口

sudo gnome-terminal

或

sudo docker container exec -it $(sudo docker ps -q)  /bin/bash

或

sudo docker ps
sudo docker container exec -it d34d89568a78  /bin/bash