实际上,Ubuntu为我们提供了可以下载NVIDIA显卡驱动的地方,打开“软件与更新”,点击“附加驱动”,此时会看到如今使用的显卡驱动是“使用X.Org X server - Nouveau display driver 来自 xserver-xorg-video-nouveau(开源)”,我们从中选择一个NVIDIA的专有驱动就好,比如我选择的是“nvidia-driver-470(专有)”,此中的470表示驱动版本,发起不要选太高,如图:
cuDNN是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。
须要根据自己的CUDA版本安装对应的cuDNN,因此须要先安装CUDA才能安装cuDNN,点击Log in | NVIDIA Developer登录后可直接进入官网,我的CUDA版本为11.4,以是我选择了CUDA版本为11.4版本对应的cuDNN,如图下载Local Installer for Linux x86_64 (Tar):