在 Ubuntu 上安装 CUDA

如果你是一名开发者，经常需要进行深度学习相关开发，那么 CUDA 是一个必不可少的工具。本文将介绍如何在 Ubuntu 系统上安装 CUDA。

步骤一：检查你的系统是否支持 CUDA

首先，你需要确定你的电脑是否支持 CUDA。在终端中运行下面的命令：

lspci | grep -i nvidia

如果你的系统返回了一些和 NVIDIA 相关的结果，那么你的系统支持 CUDA。

步骤二：下载 CUDA 安装包

你需要到 NVIDIA 官网上下载最新 CUDA 安装包，链接为 https://developer.nvidia.com/cuda-downloads 。在下载页面，选择适合你系统的安装包并下载。

步骤三：卸载旧版本的 NVIDIA 驱动

在安装新版本的 CUDA 之前，你需要卸载旧版本的 NVIDIA 驱动。在终端中运行下面的命令：

sudo apt-get remove --purge nvidia*
sudo apt autoremove

步骤四：关闭 X 服务

在安装CUDA之前，请关闭X服务。终端 Ctrl + Alt + F1 进入控制台，运行下面的命令：

sudo /etc/init.d/gdm stop

步骤五：安装 CUDA

下载完成后，解压安装包并进入解压后的文件夹。在终端中运行下面的命令：

sudo sh cuda_*.run --override

安装过程中，你需要接受 NVIDIA 的许可协议。一路默认即可。

步骤六：设置环境变量

在安装后，你需要设置 CUDA 的环境变量。在终端中运行下面的命令：

echo 'export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc

步骤七：重启电脑

重启电脑，使 CUDA 的环境变量生效。在系统登录界面按 Ctrl + Alt + F1，进入命令行界面并登录，运行：

sudo /etc/init.d/gdm start

现在，你已经成功在 Ubuntu 上安装了 CUDA。