接上回:
安装显卡驱动,一定不能使用推荐的自动安装: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

网友评论