前言
为了使用对深度学习框架进行二次开发,平台采用GPU,并且算子等需要使用CUDA,所以需要在服务器上安装GPU显卡与CUDA。
1 安装GPU驱动
- 首先确认你的显卡已经安装在你的机器上面
- 使用命令查看显示器的设备,在这里你可以看到你的显卡信息
$ sudo lshw -numeric -C display
- 然后用命令查看Ubuntu推荐的驱动版本,进行安装
$ ubuntu-drivers devices
- 我这里选择nvidia-340(这里其实应该安装384)进行安装并安装相应依赖库
$ sudo apt-get install nvidia-340 mesa-common-dev freeglut3-dev
- 安装完成后重启
$ sudo reboot
- nvidia-smi命令已经可以正常使用了,表示安装完成。
nvidia-smi
卸载nvidia驱动
因为所安装的cuda9.0要求(驱动)driver版本为384,所以需要卸载安装的340驱动,所以你在安装一系列的工具之前一定调研好每个工具要安装