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