最近再跑一个深度学习的模型,刚开始在虚拟机上跑,发现虚拟机上使用不了TensorFlow-GPU,所以速度特别的慢。决定装一个双系统使用GPU进行训练模型。
一、安装NVIDIA显卡驱动
1、通过Ubuntu上软件和更新进行安装
1.1、打开软件和更新,先更新软件源(这里我选择的是清华源),选择附加驱动,选择NVIDIA(这里我选择第一个),点击关闭,点击重新载入,安装完成。
2、手动安装NVIDIA显卡驱动
2.1、首先查看一下自己的机器显卡型号,如下图所示。
lshw -c video
2.2、知道显卡型号后,点击进入