This commit is contained in:
zeaslity
2025-05-15 10:32:14 +08:00
parent 56c79a02a8
commit ce4165e36b
93 changed files with 18779 additions and 481 deletions

View File

@@ -0,0 +1,52 @@
#!/bin/bash
download_all_dependency(){
mkdir -p /root/wdd/install/ai-run/
cd /root/wdd/install/ai-run/
wget https://oss.demo.uavcmlc.com/cmlc-installation/ai/nvidia-docker-toolkit-ubuntu.zip
wget https://oss.demo.uavcmlc.com/cmlc-installation/ai/cmlc-ai-operator_v5.4.0-v100-all.tar.gz
wget https://oss.demo.uavcmlc.com/cmlc-installation/ai/config.yaml
wget https://oss.demo.uavcmlc.com/cmlc-installation/ai/drone_detect_model_20240903.zip
}
install_nvidia_docker_runtime(){
unzip nvidia-docker-toolkit-ubuntu.zip
cd dev_files
dpkg -i libnvidia-container1_1.12.0-1_amd64.deb
dpkg -i libnvidia-container-tools_1.12.0-1_amd64.deb
dpkg -i nvidia-container-toolkit-base_1.12.0-1_amd64.deb
dpkg -i nvidia-container-toolkit_1.12.0-1_amd64.deb
dpkg -i nvidia-container-runtime_3.12.0-1_all.deb
dpkg -i nvidia-docker2_2.12.0-1_all.deb
systemctl restart docker
}
start_ai_process(){
docker stop cmlc-ai && docker rm cmlc-ai
# a100 model
docker run -itd -p2333:2333 \
--name cmlc-ai --runtime=nvidia --shm-size=8g \
--restart=always \
-v /root/wdd/install/ai-run/config.yaml:/cmii/cmlc-project-ai-streaming-engine/src/config.yaml \
harbor.cdcyy.com.cn/cmii/cmlc-ai/cmlc-ai-operator:v5.4.0-v100-all
# t4-model
docker run -itd -p2333:2333 \
--name cmlc-ai --runtime=nvidia --shm-size=8g \
--restart=always \
-v /root/wdd/ai-run/config.yaml:/cmii/cmlc-project-ai-streaming-engine/src/config.yaml \
-v /root/wdd/ai-run/drone-t4.engine:/cmii/cmlc-project-ai-streaming-engine/models/yolov8/drone.engine \
harbor.cdcyy.com.cn/cmii/cmlc-ai/cmlc-ai-operator:v5.4.0-v100-all
}
download_all_dependency

View File

@@ -0,0 +1,20 @@
^
# nvidia-docker需要安装外挂的形式
# https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
apt-get install -y nvidia-container-toolkit
nvidia-ctk runtime configure --runtime=docker
systemctl restart docker