天天看点

如何在Ubuntu上安装CUDA

原链接

在Ubuntu上安装CUDA一直是比较麻烦的事情。由于不同版本的CUDA对显卡驱动有着不同的需求。而在有些电脑上安装最新的Nvidia闭源显卡驱动会导致无法进入图形界面。CUDA中也自带了Nvidia的显卡驱动,但是如果采用默认安装,自动安装显卡驱动会导致电脑重启后无法进入图形界面。

下面就说明一下在Ubuntu上如何安装CUDA

首先安装Nvidia显卡驱动。这里采用ppa安装。

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update 
           

然后根据CUDA驱动版本要求安装驱动。也可以直接安装最新的驱动

# 2019-07-27 当前驱动最新版本
sudo apt-get install nvidia-430-dev
           

安装完成之后可以先重启一下。如果能够正常进入图形界面就说明驱动安装成功了。

接着下载和安装CUDA。注意下载时选择下载.run的文件。这个安装成功的概率更高。

安装CUDA需要关闭图形界面

按下

Ctrl + Alt + F1

进入文本界面,输入用户名密码登录后。输入

sudo service lightdm stop
           

关闭图形界面。

然后给下载的run文件添加可执行权限

# xxx是你的cuda安装文件名称
sudo chmod +x xxx.run
           

接着执行这个run文件

./xxx.run
           

根据提示进行安装,

注意默认是会安装自带的显卡驱动的,这个一定要禁用掉。否则再重启之后就无法开机了

安装完成之后根据最后的提示修改环境变量就可以了。