关于ubuntu tensorflow-gpu2.x无法使用gpu设备

最近更新系统，也顺便升级了一下显卡驱动，由原来的418升级到455，随后使用tensorflow-gpu2.1无法使用gpu加速计算，一跑网络，cpu占用率接近100%。但是gpu几乎为1%。

输入以下命令

nvidia-smi

看到驱动对应的cuda版本为11.1。而 环境中tensorflow-gpu2.1最高支持cuda10.1。需要降级显卡驱动，降级到驱动对于的cuda版本为10.1左右。

1.下载旧版本驱动，我下载的版本为430.64

2.ctrl+alt+F1 进入命令行，输入用户名和密码

3.禁用桌面服务，我的桌面为lightdm

sudo service lightdm stop

4.安装显卡驱动

sudo bash xxxxxx.run

5.启动桌面服务

sduo service lightdm restart