当前位置:实例文章 » 其他实例» [文章]Ubuntu下安装、配置及重装CUDA教程

Ubuntu下安装、配置及重装CUDA教程

发布人:shili8 发布时间:2025-01-22 23:50 阅读次数:0

**Ubuntu 下安装、配置及重装 CUDA 教程**

**前言**

CUDA 是 NVIDIA 的一个开源高性能计算库,用于在 GPU 上执行并行计算。它是机器学习、深度学习等领域的重要工具之一。在 Ubuntu 系统上安装和配置 CUDA 需要一些步骤,这里将详细介绍如何进行安装、配置及重装。

**环境准备**

* **操作系统**:Ubuntu20.04 LTS 或更高版本* **GPU**:支持 CUDA 的 NVIDIA GPU(例如 GeForce GTX1060 或更高)
* **CUDA 版本**:11.2 或更高版本**步骤一:安装必要的依赖**

首先,我们需要安装一些必要的依赖包,包括 `build-essential`、`libncurses5-dev` 和 `libstdc++6`。

bashsudo apt updatesudo apt install build-essential libncurses5-dev libstdc++6


**步骤二:下载 CUDA 安装包**

前往 NVIDIA 的官方网站,下载适合你的系统和 GPU 模型的 CUDA 安装包。例如,如果你使用 Ubuntu20.04 LTS 和 GeForce GTX1060,那么你需要下载 `cuda_11.2.1_450.51.06_linux.run`。

**步骤三:解压缩安装包**

将下载好的安装包解压缩到一个目录中,例如 `/home/user/cuda`。

bashcd /home/user/
wget  -xvf local_runtime_11.2.1_450.51.06_linux.run


**步骤四:配置 CUDA**

进入解压缩的目录,执行 `cuda` 脚本来配置 CUDA。

bashcd /home/user/cuda./cuda


按照提示输入 `yes` 或 `y` 来确认安装包的验证和安装。

**步骤五:设置环境变量**

在终端中执行以下命令来设置环境变量:

bashexport PATH=/usr/local/cuda-11.2/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH


或者,你可以将这些命令添加到你的 shell 配置文件(例如 `~/.bashrc` 或 `~/.zshrc`) 中。

**步骤六:测试 CUDA**

使用以下命令来测试 CUDA 是否正确安装:

bashnvcc --version


如果你看到 CUDA 的版本信息,那么说明安装成功了。

**重装 CUDA**

如果你需要重装 CUDA,首先卸载原来的 CUDA 安装包:

bashsudo apt remove nvidia-cuda-toolkit


然后删除 `/usr/local/cuda` 目录:

bashsudo rm -rf /usr/local/cuda


最后重新按照上面的步骤进行安装。

**总结**

在 Ubuntu 系统上安装、配置及重装 CUDA 需要一些步骤。首先,需要安装必要的依赖包,然后下载和解压缩 CUDA 安装包,接着配置 CUDA 并设置环境变量,最终测试 CUDA 是否正确安装。如果你需要重装 CUDA,那么可以按照上面的步骤进行卸载和重新安装。

**注意**

* 这个教程适用于 Ubuntu20.04 LTS 或更高版本的系统。
* 确保你的 GPU 支持 CUDA。
* 如果你遇到任何问题,请尝试在 NVIDIA 的官方论坛寻求帮助。

其他信息

其他资源

Top