关于显卡驱动安装在另一篇文章里,这里介绍Cuda+cudnn的安装
本人初学,教程中如有纰漏欢迎指正
如果需要安装,建议大家多读几篇文章作为参考,可以有效避免他人遇到的问题
如参考本篇文章,一定通读全文后再做尝试
一、首先官网查看cuda和显卡驱动版本对应关系:
(如果做学习研究用,最好提前看好自己所用架构是否能支持所选cuda版本)
二、Cuda下载地址
单击左侧Cuda版本,跳转后依次选择操作系统-计算机位数-系统型号-系统版本(由于Ubuntu系统截图修改不便,语言描述吧)
三、安装,最终参考官网吧,一步一步来安装说明
(不同版本官网说明不同,自行查找)
1.查看GPU信息:lspci | grep -i nvidia
2.查看系统信息:uname -m && cat /etc/*release
3.查看GCC信息:gcc --version
4.往下滑,找到.run文件的安装说明(Runfile Installation):
直接运行下载的.run文件安装即可:sudo sh cuda_10.1.243_418.87.00_linux.run
如果提示/tmp空间已满,莫慌,百度寻找解决办法,这里不在额外提供
继续,进入下一步
输入accept回车
取消掉Driver的勾选,安装其他的,Install回车,稍等安装结果如下:
会提示缺少命令库,不知道是否有影响,不放心的可以用如下命令安装库后,重新安装一下Cuda:sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
此时结果如下:
配置一下环境变量:sudo gedit ~/.bashrc
加入一下内容,并保存:
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
进入如下目录:cd /usr/local/cuda/samples/1_Utilities/deviceQuery
执行:sudo make
执行:sudo ./deviceQuery
四、安装Cudnn
1.下载地址官网
需要注册账号后登录,才可以下载
下载完成后,逐条执行一下命令(解压-复制文件到制定目录):
tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
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 /usr/local/cuda/lib64/libcudnn*
进行验证,执行cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
结果如下,表示安装完成
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
如需卸载,删除复制的文件即可,命令如下:
rm rf命令一定要小心谨慎使用
sudo rm -rf /usr/local/cuda-9.0/lib64/libcudnn*
sudo rm -rf /usr/local/cuda-9.0/include/cudnn.h
到此,Cuda 和Cudnn安装完成