Ubantu16.04+GPU(1070ti)驱动+cuda10.2 +CUDNN7.6.4
一、 安装nvidia显卡驱动
(亲测搭建深度学习环境,无论是win还是ubantu上的cuda里都有相对应的驱动,网上大多数教程均说单独装好驱动后在cuda,如果有那个折腾的精力可以试一试直接装cuda的时候装驱动(理论是可行的))
1、nvidia官网下载驱动
在官网里手动搜索驱动程序里,选择自己对于的系统,注意是“Linux 64-bit”。其他选项默认即可。然后,点击开始搜索,将会自动搜索你的电脑相匹配的 GPU 显卡驱动,搜索结果如下图所示:
如上图所示,结果会显示可用的 GPU 显卡驱动程序,确定你需要的版本,直接点击就可以下载了。例如我选择的版本是440.31(图是后补的哈,通常下最新的驱动即可,兼容性较好),下载后的文件是 NVIDIA-Linux-x86_64-440.31.run。
2、打开终端 Terminal,首先卸载系统之前安装的 NVIDIA 显卡驱动,尤其以前安装失败的情况下。(没有安装的可以忽略这一步)
sudo apt-get --purge remove nvidia-*
3、禁用ubantu自带驱动nouveau及其他显示驱动,非常重要打开 blacklist:
sudo gedit /etc/modprobe.d/blacklist.conf
在最后添加几行,千万注意把这些都加上,网上大多数只有一条语句 blacklist nouveau,本人亲测这样不行,估计其余显卡驱动的存在同cuda相矛盾,故安装一直失败。一定加上下述几条。
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
添加后如图:
这时候在终端输入:
lsmod | grep nouveau
如果没有输出如下图,可以直接进行下一步&#