最近实验室要在ubuntu上搭建深度学习环境,在网上查找各种搭建教程,有些不完整,踩了不少坑,在此总结一下安装过程。
这篇主要总结在Ubuntu18.04上安装NVIDIA显卡、CUDA10.1以及cuDNN7.5。
1、安装NVIDIA显卡
我这里使用的是PPA仓库自动安装NVIDIA显卡,这种方法比较简单,网上还有其他的安装方法。
首先,要将显卡驱动的PPA添加到系统中
# Add the graphics-driver PPA
$ sudo add-apt-repository ppa:graphics-drivers
# And update
$ sudo apt-get update
然后查看电脑显卡驱动信息
$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.1/0000:02:00.0 ==
modalias : pci:v000010DEd00001E04sv00007377sd00001650bc03sc00i00
vendor : NVIDIA Corporation
driver : nvidia-driver-430 - third-party free recommended
driver : nvidia-driver-410 - third-party free
driver : nvidia-driver-415 - third-party free
driver : nvidia-driver-418 - third-party free
driver : xserver-xorg-video-nouveau - distro free builtin
如果系统中有老版本显卡驱动,要先卸载