perseusdg
c0e2097397
- Updated Docker to cuda-11.3+cudnn-8.2.1+TensorRT-8.0.34,Ubuntu to 20.04 and OpenCV to 4.5.4 - Updated OpenCV4 to 4.5.4 in install_OpenCV4.sh - Updated README.md
73 lines
1.9 KiB
Bash
73 lines
1.9 KiB
Bash
#!/bin/bash
|
|
|
|
#based on https://devtalk.nvidia.com/default/topic/1042035/installing-opencv4-on-xavier/ & https://github.com/markste-in/OpenCV4XAVIER/blob/master/buildOpenCV4.sh
|
|
|
|
# Compute Capabilities can be found here https://developer.nvidia.com/cuda-gpus#compute
|
|
ARCH_BIN=7.2 # AGX Xavier
|
|
#ARCH_BIN=6.2 # Tx2
|
|
|
|
cd ~/Downloads
|
|
sudo apt-get install -y build-essential \
|
|
unzip \
|
|
pkg-config \
|
|
libjpeg-dev \
|
|
libpng-dev \
|
|
libtiff-dev \
|
|
libavcodec-dev \
|
|
libavformat-dev \
|
|
libswscale-dev \
|
|
libv4l-dev \
|
|
libxvidcore-dev \
|
|
libx264-dev \
|
|
libgtk-3-dev \
|
|
libatlas-base-dev \
|
|
gfortran \
|
|
python3-dev \
|
|
python3-venv \
|
|
libgstreamer1.0-dev \
|
|
libgstreamer-plugins-base1.0-dev \
|
|
libdc1394-22-dev \
|
|
libavresample-dev \
|
|
libtbb-dev \
|
|
|
|
git clone https://github.com/opencv/opencv.git
|
|
cd opencv && git checkout 4.5.4 && cd ..
|
|
git clone https://github.com/opencv/opencv_contrib.git
|
|
cd opencv_contrib && git checkout 4.5.4 && cd ..
|
|
|
|
|
|
python3 -m venv opencv4
|
|
source opencv4/bin/activate
|
|
pip install wheel
|
|
pip install numpy
|
|
|
|
cd opencv && mkdir build && cd build
|
|
|
|
cmake -D CMAKE_BUILD_TYPE=RELEASE \
|
|
-D CMAKE_INSTALL_PREFIX=/usr/local \
|
|
-D INSTALL_PYTHON_EXAMPLES=ON \
|
|
-D INSTALL_C_EXAMPLES=OFF \
|
|
-D OPENCV_EXTRA_MODULES_PATH='~/Downloads/opencv_contrib/modules' \
|
|
-D PYTHON_EXECUTABLE='~/Downloads/opencv4/bin/python' \
|
|
-D BUILD_EXAMPLES=ON \
|
|
-D WITH_CUDA=ON \
|
|
-D CUDA_ARCH_BIN=${ARCH_BIN} \
|
|
-D CUDA_ARCH_PTX="" \
|
|
-D ENABLE_FAST_MATH=ON \
|
|
-D CUDA_FAST_MATH=ON \
|
|
-D WITH_CUBLAS=ON \
|
|
-D WITH_LIBV4L=ON \
|
|
-D WITH_GSTREAMER=ON \
|
|
-D WITH_GSTREAMER_0_10=OFF \
|
|
-D WITH_TBB=ON \
|
|
-D WITH_OPENGL=ON \
|
|
-D WITH_VULKAN=ON \
|
|
../
|
|
|
|
make -j4
|
|
sudo make install
|
|
sudo ldconfig
|
|
|
|
cd ~/Downloads/opencv4/lib/python3.6/site-packages
|
|
ln -s /usr/local/lib/python3.6/site-packages/cv2.cpython-36m-aarch64-linux-gnu.so cv2.so
|