ubuntu20.04 apt 安装的opencv库是否支持cuda
时间: 2023-10-05 20:06:38 浏览: 155
在 Ubuntu 20.04 上,通过 apt 安装的 OpenCV 库不支持 CUDA。如果需要在 OpenCV 中使用 CUDA,需要手动编译 OpenCV 并将 CUDA 支持打开。可以参考以下步骤:
1. 安装 CUDA 工具包和驱动程序。
2. 下载 OpenCV 源代码并解压缩。
3. 安装依赖项:
```
sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
```
4. 进入 OpenCV 源代码目录并创建 build 目录:
```
cd opencv
mkdir build
cd build
```
5. 配置编译选项,并将 CUDA 支持打开:
```
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_CUDA=ON ..
```
6. 编译和安装:
```
make -j8
sudo make install
```
完成后,可以通过以下代码测试 OpenCV 是否支持 CUDA:
```
import cv2
print(cv2.cuda.getCudaEnabledDeviceCount())
```
相关问题
ubuntu20.04 apt 安装的opencv库, 使其支持cuda
要使Ubuntu 20.04中的OpenCV库支持CUDA,需要执行以下步骤:
1. 安装Nvidia显卡驱动程序和CUDA工具包:
```
sudo apt-get install nvidia-driver-450
sudo apt-get install nvidia-cuda-toolkit
```
2. 安装OpenCV库和相关依赖项:
```
sudo apt-get install libopencv-dev python3-opencv
sudo apt-get install libopencv-dev python3-opencv
```
3. 安装CUDA加速模块:
```
sudo apt-get install libopencv-dev python3-opencv
```
4. 在编译时使用CUDA加速:
```
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=7.5 -D ENABLE_FAST_MATH=1 -D CUDA_FAST_MATH=1 -D WITH_CUBLAS=1 -D WITH_LIBV4L=ON -D WITH_GSTREAMER=ON -D WITH_GTK=ON -D WITH_TBB=ON -D BUILD_opencv_python3=ON -D BUILD_opencv_python2=OFF -D BUILD_TESTS=OFF -D BUILD_PERF_TESTS=OFF -D BUILD_EXAMPLES=OFF ..
```
5. 编译和安装OpenCV库:
```
make -j8
sudo make install
```
这些步骤将使Ubuntu 20.04中的OpenCV库支持CUDA。
Ubuntu20.04 上搭建 OpenCV 开发环境
搭建 OpenCV 开发环境需要以下步骤:
1. 安装必要的依赖项
打开终端并执行以下命令:
```
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential cmake git pkg-config libgtk-3-dev \
libavcodec-dev libavformat-dev libswscale-dev libv4l-dev \
libxvidcore-dev libx264-dev libjpeg-dev libpng-dev libtiff-dev \
gfortran openexr libatlas-base-dev python3-dev python3-numpy \
libtbb2 libtbb-dev libdc1394-22-dev libopenexr-dev \
libgstreamer-plugins-base1.0-dev libgstreamer1.0-dev
```
2. 下载 OpenCV
在终端中执行以下命令:
```
cd ~
git clone https://github.com/opencv/opencv.git
cd opencv
git checkout 4.5.2
```
3. 编译和安装 OpenCV
在终端中执行以下命令:
```
cd ~/opencv
mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=RELEASE \
-D CMAKE_INSTALL_PREFIX=/usr/local \
-D INSTALL_C_EXAMPLES=ON \
-D INSTALL_PYTHON_EXAMPLES=ON \
-D OPENCV_GENERATE_PKGCONFIG=YES \
-D WITH_TBB=ON \
-D WITH_V4L=ON \
-D WITH_QT=OFF \
-D WITH_OPENGL=ON \
-D WITH_CUDA=OFF \
-D BUILD_EXAMPLES=ON ..
```
编译后,在终端中执行以下命令进行安装:
```
make -j$(nproc)
sudo make install
sudo ldconfig
```
4. 验证 OpenCV 安装
创建一个名为 `test.cpp` 的文件,内容如下:
```
#include <opencv2/opencv.hpp>
using namespace cv;
int main() {
Mat img = imread("test.jpg");
imshow("test", img);
waitKey(0);
return 0;
}
```
将 `test.jpg` 文件放在同一个目录下,然后在终端中执行以下命令:
```
g++ test.cpp -o test `pkg-config --cflags --libs opencv4`
./test
```
如果一切正常,应该会看到 `test.jpg` 文件被打开并显示在屏幕上。
以上就是在 Ubuntu20.04 上搭建 OpenCV 开发环境的步骤。
阅读全文