美文网首页
ubuntu opencv-python gpu 加速记录(2)

ubuntu opencv-python gpu 加速记录(2)

作者: 出海啊 | 来源:发表于2022-09-05 20:23 被阅读0次

    接上回:

    安装显卡驱动,一定不能使用推荐的自动安装:sudo ubuntu-drivers autoinstall

    因为他会把你的网卡驱动,或者其他给搞没。

    而是要选择:sudo apt install nvidia-utils-470-server

    这种方式,安装需要的制定版本。

    在终端输入:ubuntu-drivers devices,可以看到需要什么版本。


    安装:cuda

    sudo apt install nvidia-cuda-toolkit

    大概耗时10分钟........



    综合以上:

    1、显卡驱动

    2、nvcc

    都好了,开始编译opencv-python


    卸载已经安装的 opencv 

    pip list 查找

    pip uninstall opencv-python*


    准备编译带GPU的opencv

    下载opencv源码:

    https://github.com/Itseez/opencv

    安装编译依赖:

    sudo apt-get update && sudo apt-get install -y    build-essential cmake git libgtk2.0-dev pkg-config    libavcodec-dev libavformat-dev libswscale-dev python3-dev    python3-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev


    开始编译:

    sudo cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local -D BUILD_opencv_python2=OFF -D BUILD_opencv_python3=ON -D WITH_CUDA=ON -D OPENCV_EXTRA_MODULES_PATH=/home/mike/opencv/opencv-contrib/modules -D PYTHON3_EXECUTABLE=/home/mike/miniconda3/envs/ios/bin/python -D CUDA_NVCC_FLAGS=--expt-relaxed-constexpr ..

    OPENCV_EXTRA_MODULES_PATH 用来存放opencv-contrib

    PYTHON3_EXECUTABLE    指定python

    删掉缓存:CMakeCache.txt

    相关文章

      网友评论

          本文标题:ubuntu opencv-python gpu 加速记录(2)

          本文链接:https://www.haomeiwen.com/subject/trtbnrtx.html