ubuntu安装支持cuda的opencv
时间: 2023-11-15 20:04:53 浏览: 141
好的,以下是安装支持CUDA的OpenCV的步骤:
1. 安装CUDA和cuDNN
首先需要安装CUDA和cuDNN,可以参考官方文档进行安装。
2. 安装依赖库
sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libdc1394-22-dev
3. 下载OpenCV源码
可以从OpenCV官网下载最新版本的源码。
4. 编译安装OpenCV
进入OpenCV源码目录,执行以下命令:
mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_CUDA=ON -D CUDA_ARCH_BIN=7.5 -D CUDA_ARCH_PTX="" -D WITH_CUBLAS=ON -D ENABLE_FAST_MATH=1 -D CUDA_FAST_MATH=1 -D WITH_CUDNN=ON -D OPENCV_DNN_CUDA=ON -D BUILD_opencv_cudacodec=OFF -D BUILD_opencv_world=OFF ..
make -j8
sudo make install
其中,CUDA_ARCH_BIN参数需要根据自己的显卡型号进行设置,7.5是适用于GTX 1080 Ti等显卡的设置。
5. 验证安装结果
在终端中输入python,进入Python交互环境,输入以下代码:
import cv2
print(cv2.cuda.getCudaEnabledDeviceCount())
如果输出结果大于0,则说明安装成功。