ubuntu系统安装CUDA【笔记】

实践设备为：华硕FX-PRO（NVIDIA GeForce GTX 960M）

在Ubuntu系统上安装CUDA主要分为以下几个步骤：

1. 准备工作

首先，你需要确认你的Ubuntu系统版本和你的NVIDIA显卡型号。因为不同的Ubuntu版本和NVIDIA显卡可能需要不同版本的CUDA。你可以通过lscpu命令查看你的CPU信息，通过nvidia-smi命令查看你的NVIDIA显卡信息。

2. 安装NVIDIA驱动

如果你的Ubuntu系统还没有安装NVIDIA驱动，你需要先安装NVIDIA驱动。你可以通过Ubuntu系统的软件管理器或者使用命令安装对应的NVIDIA驱动，具体安装方法可参考《UEFI安全启动模式下安装Ubuntu的NVIDIA显卡驱动》。

3. 安装CUDA

安装完NVIDIA驱动之后，你就可以开始安装CUDA了。你可以从NVIDIA官方网站下载对应版本的CUDA安装包，这里以CUDA Toolkit 12.4为例。

下载完成之后，你可以使用sudo sh cuda_xx.x_xxxxxx_linux.run命令安装CUDA。

安装完成。

4. 配置环境变量

安装完CUDA之后，你可能需要配置环境变量。你可以编辑你的~/.bashrc文件，添加以下代码：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-xx.x/lib64
export PATH=$PATH:/usr/local/cuda-xx.x/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-xx.x

其中xx.x是你的CUDA版本号。然后，你可以使用source ~/.bashrc命令使环境变量的改动生效。

可以使用nvcc -V命令来验证CUDA是否安装成功。如果输出CUDA的版本信息，则说明安装成功。

以上就是在Ubuntu系统上安装CUDA的主要步骤，希望对你有所帮助。

申明： 
所有资料为本人收集整理，仅限个人学习使用，勿做商业用途。