系统:Ubuntu16.04或ubuntu18.04 的 虚机/物理机

GPU型号:GeForce/Tesla/等

 

拢共有两种方法

A)通过ubuntu系统提供工具安装

B)通过官网下载驱动手动安装

 

安装的原则是,越简单越好

【通过ubuntu系统提供工具安装】是优选,安装失败再考虑手动安装

 

 

安装之前要明确的注意事项

1)确定自己要安装的驱动版本

驱动限制cuda,cuda又限制tensorflow_gpu版本,所以安装的时候尽量选择最高版本的驱动,cuda及cudnn一般是多版本共存,靠软连接分情况切换,如果没别的要求,就选择ubuntu系统推荐的最高版本

不知道的可以参考:https://blog.csdn.net/mouse1598189/article/details/86695400

2)先装驱动再装cuda 然后再cudnn - anaconda - tensorflow

 

开始!

 

0-卸载已安装驱动

sudo apt-get purge nvidia*

A选项-【通过ubuntu系统提供工具安装】

A1-更新

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

查看支持版本

ubuntu-drivers devices

A2-安装

sudo apt-get install nvidia-384

B选项-【通过官网下载驱动手动安装】

B1-下载驱动:https://www.nvidia.cn/Download/index.aspx?lang=cn

选择相应版本下载

B2-禁用nouveau

编辑黑名单

sudo gedit /etc/modprobe.d/blacklist.conf

末尾添加:

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

B3-重启

sudo update-initramfs -u
sudo reboot

检测是否禁用成功

lsmod | grep nouveau

没有返回就对了

 

B4-进入tty控制台(Ctrl + alt + F1)**重要**

B5-关闭桌面

sudo service lightdm stop

B6-安装刚刚下载的驱动

sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files

B7-开启桌面

sudo service lightdm start

结束

测试:

$ nvidia-smi

输出:

 

 

其他问题:

1)多cuda情况:

安装cuda的时候一定注意:问你是否创建软连接/usr/local/cuda 一定不要选!!!!否则电脑重启进不去ubuntu桌面。

2)出现下面warning,么有事ok继续就行

3)sudo service lightdm stop出错

输出:unit lightdm.service not loaded

 

原因:没有安装lightdm 安装就是了

 sudo apt install lightdm

选择部分就选lightdm

 

6)GCC、G++版本太低安装也会失败,多备份几个,报错就切换
 

 

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐