ubuntu安装显卡驱动，cuda及cudnn

ubuntu安装显卡驱动，cuda及cudnn

1.安装显卡驱动

首先在终端输入
$ nvidia-smi (检查显卡驱动)，若有则出现注：以上显卡驱动安装的是440版本。
若没有出现上述图片，则继续在终端输入
$ ubuntu-drivers devices（作用是检测你的NVIDIA显卡型号和推荐的驱动程序的模型）
以上是可安装的驱动，这里安装推荐的驱动，为nvidia-driver-440。
继续在终端输入：
$ sudo apt-get install nvidia-driver-440
安装完成后一定要重启电脑，然后在终端输入nvidia-smi出现上述第一张图片说明安装成功。

2.安装cuda

虽然显卡驱动支持cuda10.2版本，但还是要根据自己用的框架进行选择cuda版本，这里用的是pytorch，所以根据官网的推荐是用cuda10.1，下面安装cuda10.1。
cuda下载地址：下载地址

按照官网指令安装：
注意选择安装选项时：不选驱动

设置cuda环境变量

输入 sudo gedit ~/.bashrc 打开主目录下的 .bashrc文件添加如下路径:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1 # 根据自己的目录来

终端运行：source ~/.bashrc 使之生效；
检查安装是否成功：

输入 nvcc --version，如果显示如下文字就说明安装成功了：

3.cudnn安装

自己百度cudnn进官网登录下载
选择适合的版本下载（具体可以看最后参考资料）。
安装好后解压。拷贝.h 和 libs文件到cuda安装目录，并给予执行权限：

$ 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*　＃　具体看自己目录

最后检测下cuda是否能用：
使用cuda文件夹中的测试文件,

$ cd /usr/local/cuda/samples/1_Utilities/deviceQuery
$ sudo ./deviceQuery

结果出现result=pass则成功。

最后在自己安装好的pytorch环境下检测cuda：
torch.cuda.is_available()
True
成功。

参考资料

https://blog.csdn.net/sss_369/article/details/94591280

https://blog.csdn.net/weixin_45024901/article/details/89893338