如果要安装带CUDA的OpenCV,先按照本篇博客安装CUDA,然后参考我另一篇博客Ubuntu 编译OpenCV安装OpenCV。
一、安装驱动
其实Ubuntu 16.04就附带NVIDIA显卡驱动,只是默认没启用而已。自带的驱动稳定的多,不存在奇奇怪怪的问题。
- 打开“系统设置(System Settings)”-->“软件和更新(Software & Updates)”
- 选择“附加驱动(Addtional Drivers)”,等待几分钟,出现如下界面
- 然后勾选“使用NVIDIDA...(Using NVIDIA...)”,点击“应用更改”,然后输入密码,等待半小时左右就可以完成驱动的安装。
二、下载CUDA安装文件
Ubuntu 16.04的自带的驱动支持CUDA 8.0 和CUDA 9.0;Ubuntu 18.04自带的驱动支持CUDA 9.0和CUDA 9.1。
Ubuntu 16.04也可以安装CUDA 9.1,不过需要更新驱动,可参考其他博客。不过不建议在Ubuntu 16.04上安装CUDA 9.1。
OpenCV2.4.13仅支持CUDA 8.0;OpenCV 3.2.0仅支持CUDA8.0;OpenCV 3.4.2支持CUDA 9.1、CUDA 9.0、CUDA 8.0。
所以如果你使用Ubuntu16.04,那么安装CUDA 8.0的时候OpenCV版本可以随意选;如果安装的是CUDA 9.0,那么OpenCV得选3.4.0以上了(2.4.*和3.2.*不行,3.3.*没有测试过)。如果你使用Ubuntu 18.04,那么必须安装CUDA 9.1,OpenCV得选3.4.0以上了(2.4.*和3.2.*不行,3.3.*没有测试过)。
- CUDA下载汇总地址
https://developer.nvidia.com/cuda-toolkit-archive
- CUDA 8.0下载地址
https://developer.nvidia.com/cuda-80-ga2-download-archive
进入下载页面后,向下图一样选择
然后把“Base Installer”和“Parch2”下载下来。
- CUDA 9.1下载地址
https://developer.nvidia.com/cuda-91-download-archive
进入下载页面后,做如下选择
因为CUDA还没有提供Ubuntu 18.04的安装包,所以可以下载17.04的安装包,经过测试,在Ubuntu18.04一样可以用。 把然后把“Base Installer”和“Parch1”~“Parch3”下载下来。
三、安装CUDA
在Ubuntu 16.04上安装CUDA8.0和CUDA9.0就只是需要多下载安装几个补丁包(8.0只有一个、9.0有四个),所以本篇博客只贴出Ubuntu 16.04安装CUDA 8.0的和Ubuntu 18.04安装CUDA 9.1。
- 添加执行权限
CUDA 8.0
chmod +x cuda_8.0.61_375.26_linux.run
chmod +x cuda_8.0.61.2_linux.run
CUDA 9.1
chmod +x cuda_9.1.85_387.26_linux.run
chmod +x cuda_9.1.85.1_linux.run
chmod +x cuda_9.1.85.2_linux.run
chmod +x cuda_9.1.85.3_linux.run
- Ubuntu 16.04安装CUDA 8.0
打开终端,输入
sudo ./cuda_8.0.61_375.26_linux.run
然后一直按F, 然后输入“accept”接受协议;当询问你是否安装驱动的时候,要输入"n",因为之前就安装过驱动了;主要的选项如下图
然后再安装补丁包
sudo ./cuda_8.0.61.2_linux.run
- Ubuntu 18.04安装CUDA 9.1
打开终端,输入
sudo ./cuda_9.1.85_387.26_linux.run
然后一直按F, 然后输入“accept”接受协议;因为CUDA 9.1不是为18.04编译的,所以会提示你没有给这个平台编译balabala,输入“y”就行;当询问你是否安装驱动的时候,要输入"n",因为之前就安装过驱动了;主要的选项如下图
然后依次安装补丁包
sudo ./cuda_9.1.85.1_linux.run
sudo ./cuda_9.1.85.2_linux.run
sudo ./cuda_9.1.85.3_linux.run
- 添加环境变量
打开终端,输入”gedit ~/.bashrc“,然后再弹出的编辑器的最后,添加如下两行
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
保存退出。
四、测试CUDA
进入你的CUDA Example所在目录,默认是主目录,找到“NVIDIA_CUDA-8.0_Samples”或者“NVIDIA_CUDA-9.1_Samples”。依次打开“1_Utilities”-->"deviceQuery",然后打开终端输入
source ~/.bashrc # 重启之后就不用再执行这一行了
make
./deviceQuery
然后出现”Result = PASS“字样时,说明安装成功了。