cuda ubuntu安装_ubuntu配置英伟达实现GPU加速

卸载系统里低版本的英伟达驱动

sudo apt-get purge nvidia*

把显卡驱动加入PPA

sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update

查找英伟达显卡驱动最新版本号

sudo apt-cache search nvidia

查看Ubuntu驱动

使用终端命令查看系统推荐的驱动版本

ubuntu-drivers devices

然后打开系统里的软件和更新,点击附加驱动,选择相应的NVIDIA进行应用更改.

CUDA安装

打开英伟达官网

https://developer.nvidia.com/cuda-toolkit-archive

然后选择下载合适的版本下载即可.

设置环境变量

打开主目录下的.bashrc文件添加如下路径,则按Ctrl+H键显示隐藏文件.

以下命令中cuda-10.2需要看个人情况修改.

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2

检查

nvcc --version

如果显示相关版本就说明安装成功了.

cudnn安装

打开官网

https://developer.nvidia.com/rdp/cudnn-archive

选择图中的CUDNN

点击下载按钮,然后显示登录英伟达账号,注册一个即可,然后就会出现以下界面,根据需要下载即可.

安装cudnn

解压下载的文件,在当前目录打开终端,执行如下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
 
sudo chmod a+r /usr/local/cuda/include/cudnn.h
 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

查看cudnn版本

在终端输入

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

出现版本信息,说明安装好了.

本文一些图片资料可参考我的公众号如下文章链接

Ubuntu配置英伟达进行GPU加速​mp.weixin.qq.com