forked from floratest1/SpireCV
upd install scripts
This commit is contained in:
Your Name
@@ -0,0 +1,39 @@
|
||||
#!/bin/bash -e
|
||||
|
||||
root_dir=${HOME}"/SpireCV/models"
|
||||
root_server="https://download.amovlab.com/model"
|
||||
|
||||
sv_params1=${HOME}"/SpireCV/sv_algorithm_params.json"
|
||||
sv_params2=${HOME}"/SpireCV/sv_algorithm_params_coco_640.json"
|
||||
sv_params3=${HOME}"/SpireCV/sv_algorithm_params_coco_1280.json"
|
||||
camera_params1=${HOME}"/SpireCV/calib_webcam_640x480.yaml"
|
||||
camera_params2=${HOME}"/SpireCV/calib_webcam_1280x720.yaml"
|
||||
|
||||
|
||||
if [ ! -d ${root_dir} ]; then
|
||||
echo -e "\033[32m[INFO]: ${root_dir} not exist, creating it ... \033[0m"
|
||||
mkdir -p ${root_dir}
|
||||
fi
|
||||
|
||||
if [ ! -f ${sv_params1} ]; then
|
||||
echo -e "\033[32m[INFO]: ${sv_params1} not exist, downloading ... \033[0m"
|
||||
wget -O ${sv_params1} ${root_server}/install/a-params/sv_algorithm_params.json
|
||||
fi
|
||||
if [ ! -f ${sv_params2} ]; then
|
||||
echo -e "\033[32m[INFO]: ${sv_params2} not exist, downloading ... \033[0m"
|
||||
wget -O ${sv_params2} ${root_server}/install/a-params/sv_algorithm_params_coco_640.json
|
||||
fi
|
||||
if [ ! -f ${sv_params3} ]; then
|
||||
echo -e "\033[32m[INFO]: ${sv_params3} not exist, downloading ... \033[0m"
|
||||
wget -O ${sv_params3} ${root_server}/install/a-params/sv_algorithm_params_coco_1280.json
|
||||
fi
|
||||
|
||||
if [ ! -f ${camera_params1} ]; then
|
||||
echo -e "\033[32m[INFO]: ${camera_params1} not exist, downloading ... \033[0m"
|
||||
wget -O ${camera_params1} ${root_server}/install/c-params/calib_webcam_640x480.yaml
|
||||
fi
|
||||
if [ ! -f ${camera_params2} ]; then
|
||||
echo -e "\033[32m[INFO]: ${camera_params2} not exist, downloading ... \033[0m"
|
||||
wget -O ${camera_params2} ${root_server}/install/c-params/calib_webcam_1280x720.yaml
|
||||
fi
|
||||
|
||||
@@ -0,0 +1,66 @@
|
||||
#!/bin/sh
|
||||
|
||||
|
||||
sudo apt install -y \
|
||||
build-essential yasm cmake libtool libc6 libc6-dev unzip wget libeigen3-dev libfmt-dev \
|
||||
libnuma1 libnuma-dev libx264-dev libx265-dev libfaac-dev libssl-dev
|
||||
|
||||
current_dir=$(pwd)
|
||||
root_dir=${HOME}"/SpireCV"
|
||||
if [ ! -d ${root_dir} ]; then
|
||||
echo -e "\033[32m[INFO]: ${root_dir} not exist, creating it ... \033[0m"
|
||||
mkdir -p ${root_dir}
|
||||
fi
|
||||
cd ${root_dir}
|
||||
|
||||
git clone https://gitee.com/jario-jin/nv-codec-headers.git
|
||||
cd nv-codec-headers
|
||||
git checkout n11.1.5.0
|
||||
sudo make install
|
||||
cd ..
|
||||
sudo rm -r nv-codec-headers
|
||||
|
||||
wget https://ffmpeg.org/releases/ffmpeg-4.2.5.tar.bz2
|
||||
tar -xjf ffmpeg-4.2.5.tar.bz2
|
||||
cd ffmpeg-4.2.5
|
||||
export PATH=$PATH:/usr/local/cuda/bin
|
||||
sed -i 's#_30#_75#' configure; sed -i 's#_30#_75#' configure
|
||||
./configure \
|
||||
--enable-nonfree \
|
||||
--enable-gpl \
|
||||
--enable-shared \
|
||||
--enable-ffmpeg \
|
||||
--enable-ffplay \
|
||||
--enable-ffprobe \
|
||||
--enable-libx264 \
|
||||
--enable-libx265 \
|
||||
--enable-cuda-nvcc \
|
||||
--enable-nvenc \
|
||||
--enable-cuda \
|
||||
--enable-cuvid \
|
||||
--enable-libnpp \
|
||||
--extra-libs="-lpthread -lm" \
|
||||
--extra-cflags=-I/usr/local/cuda/include \
|
||||
--extra-ldflags=-L/usr/local/cuda/lib64
|
||||
make -j8
|
||||
sudo make install
|
||||
cd ..
|
||||
|
||||
git clone https://gitee.com/jario-jin/ZLMediaKit.git
|
||||
cd ZLMediaKit
|
||||
git submodule update --init
|
||||
mkdir build
|
||||
cd build
|
||||
cmake ..
|
||||
make -j4
|
||||
cd ..
|
||||
cd ..
|
||||
|
||||
mkdir ZLM
|
||||
cd ZLM
|
||||
cp ../ZLMediaKit/release/linux/Debug/MediaServer .
|
||||
cp ../ZLMediaKit/release/linux/Debug/config.ini .
|
||||
cd ..
|
||||
|
||||
cd ${current_dir}
|
||||
|
||||
+2
-1
@@ -5,7 +5,7 @@ wget https://download.amovlab.com/model/deps/opencv-4.7.0.zip
|
||||
wget https://download.amovlab.com/model/deps/opencv_contrib-4.7.0.zip
|
||||
wget https://download.amovlab.com/model/deps/opencv_cache_x86-4.7.0.zip
|
||||
|
||||
|
||||
current_dir=$(pwd)
|
||||
package_dir="."
|
||||
mkdir ~/opencv_build
|
||||
|
||||
@@ -67,3 +67,4 @@ sudo make install
|
||||
|
||||
cd
|
||||
sudo rm -r ~/opencv_build
|
||||
cd ${current_dir}
|
||||
@@ -5,7 +5,7 @@ wget https://download.amovlab.com/model/deps/opencv-4.7.0.zip
|
||||
wget https://download.amovlab.com/model/deps/opencv_contrib-4.7.0.zip
|
||||
wget https://download.amovlab.com/model/deps/opencv_cache_x86-4.7.0.zip
|
||||
|
||||
|
||||
current_dir=$(pwd)
|
||||
package_dir="."
|
||||
mkdir ~/opencv_build
|
||||
|
||||
@@ -67,3 +67,4 @@ sudo make install
|
||||
|
||||
cd
|
||||
sudo rm -r ~/opencv_build
|
||||
cd ${current_dir}
|
||||
@@ -0,0 +1,95 @@
|
||||
#!/bin/sh
|
||||
|
||||
echo "\033[32m[INFO]:\033[0m Please enter the folder path of the installation package: "
|
||||
# package_dir="/home/jario/Downloads/nv"
|
||||
|
||||
wget https://download.amovlab.com/model/install/x86-nvidia/cuda-repo-ubuntu1804-11-1-local_11.1.1-455.32.00-1_amd64.deb
|
||||
wget https://download.amovlab.com/model/install/x86-nvidia/cudnn-11.3-linux-x64-v8.2.1.32.tgz
|
||||
wget https://download.amovlab.com/model/install/x86-nvidia/nv-tensorrt-repo-ubuntu1804-cuda11.3-trt8.0.1.6-ga-20210626_1-1_amd64.deb
|
||||
wget https://download.amovlab.com/model/install/x86-nvidia/cuda-ubuntu1804.pin
|
||||
|
||||
package_dir="."
|
||||
|
||||
cuda_fn=$package_dir"/cuda-repo-ubuntu1804-11-1-local_11.1.1-455.32.00-1_amd64.deb"
|
||||
cudnn_fn=$package_dir"/cudnn-11.3-linux-x64-v8.2.1.32.tgz"
|
||||
tensorrt_fn=$package_dir"/nv-tensorrt-repo-ubuntu1804-cuda11.3-trt8.0.1.6-ga-20210626_1-1_amd64.deb"
|
||||
tmp_dir="/tmp"
|
||||
|
||||
echo "\033[32m[INFO]: CUDA_PKG: \033[0m"$cuda_fn
|
||||
echo "\033[32m[INFO]: CUDNN_PKG: \033[0m"$cudnn_fn
|
||||
echo "\033[32m[INFO]: TENSORRT_PKG: \033[0m"$tensorrt_fn
|
||||
|
||||
# 所有文件都存在时，才会继续执行脚本
|
||||
if [ ! -f "$cuda_fn" ]; then
|
||||
echo "\033[31m[ERROR]: CUDA_PKG not exist!: \033[0m"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ ! -f "$cudnn_fn" ]; then
|
||||
echo "\033[31m[ERROR]: CUDNN_PKG not exist!: \033[0m"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ ! -f "$tensorrt_fn" ]; then
|
||||
echo "\033[31m[ERROR]: TENSORRT_PKG not exist!: \033[0m"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# 删除显卡驱动
|
||||
# sudo apt-get remove nvidia-*
|
||||
|
||||
# 安装显卡驱动
|
||||
# echo "\033[32m[INFO]: Nvidia Driver installing ...\033[0m"
|
||||
# ubuntu-drivers devices
|
||||
# sudo ubuntu-drivers autoinstall
|
||||
|
||||
# 删除已安装CUDA
|
||||
# --purge选项会将配置文件、数据库等删除
|
||||
# sudo apt-get autoremove --purge cuda
|
||||
# 查看安装了哪些cuda相关的库，可以用以下指令
|
||||
# sudo dpkg -l |grep cuda
|
||||
# sudo dpkg -P cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01
|
||||
# sudo dpkg -P cuda-repo-ubuntu1804-11-1-local
|
||||
# sudo dpkg -P nv-tensorrt-repo-ubuntu1804-cuda10.2-trt8.0.1.6-ga-20210626
|
||||
# 这个key值是官网文档查到的，当然通过apt-key list也能查看
|
||||
# sudo apt-key list
|
||||
# sudo apt-key del 7fa2af80
|
||||
|
||||
# 安装CUDA
|
||||
echo "\033[32m[INFO]: CUDA installing ...\033[0m"
|
||||
# wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
|
||||
sudo cp $package_dir/cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
|
||||
sudo dpkg -i $cuda_fn
|
||||
sudo apt-key add /var/cuda-repo-ubuntu1804-11-1-local/7fa2af80.pub
|
||||
sudo apt-get update
|
||||
sudo apt-get -y install cuda
|
||||
|
||||
# 安装CUDNN
|
||||
echo "\033[32m[INFO]: CUDNN installing ...\033[0m"
|
||||
tar zxvf $cudnn_fn -C $tmp_dir
|
||||
sudo cp $tmp_dir/cuda/include/cudnn* /usr/local/cuda/include/
|
||||
sudo cp $tmp_dir/cuda/lib64/libcudnn* /usr/local/cuda/lib64/
|
||||
sudo chmod a+r /usr/local/cuda/include/cudnn* /usr/local/cuda/lib64/libcudnn*
|
||||
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn.so.8
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_ops_train.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_ops_train.so.8
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_adv_train.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_adv_train.so.8
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8
|
||||
sudo ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8.2.1 /usr/local/cuda/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8
|
||||
|
||||
# 安装TensorRT
|
||||
sudo dpkg -i $tensorrt_fn
|
||||
sudo apt-key add /var/nv-tensorrt-repo-cuda11.3-trt8.0.1.6-ga-20210626/7fa2af80.pub
|
||||
sudo apt-get update
|
||||
sudo apt-get install tensorrt -y
|
||||
sudo apt-get install python3-libnvinfer-dev -y
|
||||
|
||||
sudo rm $cuda_fn
|
||||
sudo rm $cudnn_fn
|
||||
sudo rm $tensorrt_fn
|
||||
sudo rm cuda-ubuntu1804.pin
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user