Ubuntu下的CUDA + OPENCL安装

CUDA和OPENCL安装

前言:最近需要做一些GPU并行计算,因而入坑CUDA和OPENCL,两者都有用到一些,刚好有点时间,同时记录一些学习过程,排掉一些坑,这篇是环境安装篇,基本跟着走就没什么问题。

环境:ubuntu18.04 / ubuntu20.04 显卡:Nvidia

CUDA安装

1.查看电脑是否识别GPU
lspci | grep -i nvidia

在这里插入图片描述

2.查看电脑可以安装的显卡驱动版本
  ubuntu-drivers devices

在这里插入图片描述

3.安装显卡驱动,选择上图的recommended
sudo apt install nvidia-driver-XXX
reboot #重启
4.查看显卡信息
nvidia-smi

在这里插入图片描述

5.安装CUDA

到nivida官网下载自己可安装的版本

在这里插入图片描述

复制Base Installer的执行代码,下载脚本(3个多G,可能有点久)

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run #选择自己的版本

运行脚本进行安装cuda库

sudo sh cuda_11.4.0_470.42.01_linux.run

在弹出的安装界面中选“continue”,如果选了会跳出安装,就说明安装失败,给了失败日志的路径,可以自己查看原因。
在这里插入图片描述

进入下一步,输入accept

在这里插入图片描述

把Driver选项的X去掉,不然它会再给你装一次驱动,会冲突,报错。然后选择Install

在这里插入图片描述

等待安装,没什么问题就成功了。

6.配置环境变量
gedit ~/.bashrc

最后一行增加

export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

注意查看自己的cuda真是安装路径

7.检查是否安装成功

重新打开终端

nvcc -V

在这里插入图片描述

8.添加cuda lib 路径

注意自己的版本号

sudo echo '/usr/local/cuda-11.4/lib64/' >> /etc/ld.so.conf
sudo ldconfig

CUDA安装成功之后会在终端当前所在目录库生成一个NVIDIA_CUDA-11.4_Samples的编程例子的文件夹,可以用来测试。直接进去目录里面make就行。

OPENCL安装

1.安装

其实CUDA安装是包含了OPENCL的,但是你要使用OPENCL还需要配置一下。

也可以单独安装头文件和库

sudo apt install opencl-headers 
sudo apt install ocl-icd-libopencl1 
sudo apt install ocl-icd-opencl-dev
sudo apt install clinfo
#sudo apt install intel-opencl-icd  #Install NEO OpenCL runtime for Intel GPU

查看电脑当前安装信息

clinfo

  • 1
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
可以直接安装OpenCV 4.x或者OpenCV 3.x,具体步骤可以在终端中运行以下命令: 1. 更新系统软件包 sudo apt-get update && sudo apt-get upgrade 2. 安装OpenCV依赖 sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev libv4l-dev libxvidcore-dev libx264-dev libjpeg-dev libpng-dev libtiff-dev gfortran openexr libatlas-base-dev python-dev python-bumpy libgstreamer-plugins-base1.0-dev libavresample-dev libgphoto2-dev libopenblas-dev libwebp-dev libatlas-base-dev liblapack-dev libblas-dev libeigen3-dev libhdf5-dev libprotobuf-dev protobuf-compiler libgoogle-glog-dev libgflags-dev libtbb2 libtbb-dev libdc1394-22-dev libvorbis-dev libogg-dev libtheora-dev libx264-dev libxvidcore-dev libfaac-dev libmp3lame-dev libopencore-amrnb-dev libopencore-amrwb-dev libvpx-dev libprotobuf-dev protobuf-compiler libgoogle-glog-dev libgflags-dev libtesseract-dev -y 3. 下载OpenCV源码 cd ~ git clone https://github.com/opencv/opencv.git cd opencv git checkout 4.0.0 4. 编译并安装OpenCV cd ~ cd opencv mkdir build cd build cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D BUILD_NEW_PYTHON_SUPPORT=ON -D BUILD_opencv_python3=ON -D INSTALL_PYTHON_EXAMPLES=ON -D INSTALL_C_EXAMPLES=OFF -D OPENCV_ENABLE_NONFREE=ON -D WITH_CUDA=OFF -D WITH_CUDNN=OFF -D WITH_OPENGL=ON -D WITH_OPENCL=ON -D WITH_IPP=ON -D WITH_TBB=ON -D WITH_EIGEN=ON -D WITH_DNN=ON -D WITH_FFMPEG=ON -D WITH_GSTREAMER=ON -D WITH_GSTREAMER_0_10=OFF -D WITH_OPENEXR=ON -D OPENCV_EXTRA_MODULES_PATH=~/opencv_contrib/modules -D PYTHON3_LIBRARY=/usr/lib/x86_64-linux-gnu/libpython3.6m.so -D PYTHON3_INCLUDE_DIR=/usr/include/python3.6m -D PYTHON3_EXECUTABLE=/usr/bin/python3 -D BUILD_EXAMPLES=ON .. make -j8 sudo make install 安装完成后,你可以尝试使用以下命令测试是否安装成功: python3 import cv2 print(cv2.__version__)

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值