当前位置: 乱七八糟 » Ubuntu » ubuntu18.04安装nvidia CUDA方法及可能出现的问题

ubuntu18.04安装nvidia CUDA方法及可能出现的问题

近两年人工智能机器学习大热,很多人在接触学习机器学习,nvidia cuda 能够加速深度学习的模型训练,但不少人在安装CUDA的时候遇到各种问题。下面是在ubuntu18.04下安装cuda的官方方法。

驱动问题:

用NVIDIA官方的方法, 直接安装cuda既可,在安装过程中会自动安装、更新驱动。

安装方法:

在NVIDIA官网下载安装软件包,并且直接根据下方的安装步骤走即可。


下载地址:https://developer.nvidia.com/cuda-downloads

sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

安装建议:

建议选择local deb包的方式安装, 试过deb network的方式,没能简单的安装上。

其它问题:

安装完之后,ubuntu出现登陆无法显示桌面环境的问题(登陆界面的图形界面可以显示),可能是安装的过程中把X服务的部分破坏了,可以尝试重新安装xorg及所用的桌面环境(gnome)。不过如果是用作远程操作的话,这个就没有什么区别的,卸载了桌面环境更省内存。