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 的官方论坛寻求帮助。