超简单:Linux下opencv-gpu配置

1.下载opencv和opencv_contrib安装包
1)使用命令下

git clone https://github.com/opencv/opencv.git -b 4.9.0
git clone https://github.com/opencv/opencv_contrib.git -b 4.9.0

2)复制链接去GitHub下载然后上传到服务器
超简单:Linux下opencv-gpu配置_第1张图片
注意:看好版本,opencv和opencv_contrib安装包的版本必须一致。

2.配置 CMake

cd opencv && mkdir build && cd build
cmake \
  -D CMAKE_BUILD_TYPE=RELEASE \
  -D CMAKE_INSTALL_PREFIX=/usr/local \ #因为我是再docker配置的,所以/usr/local 
  -D WITH_CUDA=ON \
  -D CUDA_ARCH_BIN=7.0 \  # V100 的计算能力为 7.0,这横自行需改
  -D CUDA_FAST_MATH=ON \
  -D WITH_CUDNN=ON \
  -D OPENCV_DNN_CUDA=ON \
  -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
  -D WITH_NVCUVID=ON \
  -D BUILD_EXAMPLES=OFF \
  ..

注意:在编译过程中遇到什么下载失败,如下图所示不需要管。
超简单:Linux下opencv-gpu配置_第2张图片
3.编译与安装

make -j$(nproc)
sudo make install

同样,遇到网络问题不用管,因为我们不需要下载案例,只是想使用opencv-gpu。
4.验证

python3
import cv2
print(cv2.getBuildInformation())  # 检查输出中是否有 "CUDA: YES"
print(cv2.cuda.getCudaEnabledDeviceCount())  # 应返回 1(表示检测到 GPU)

超简单:Linux下opencv-gpu配置_第3张图片

你可能感兴趣的:(学习笔记,linux,opencv,运维)