当前位置:首页 > 系统教程 > 正文

Linux环境下安装CUDA 11.8全攻略(新手必看的深度学习环境配置指南)

Linux环境下安装CUDA 11.8全攻略(新手必看的深度学习环境配置指南)

在开展深度学习或高性能计算工作时,Linux安装CUDA 11.8是构建开发环境的核心步骤。CUDA(Compute Unified Device Architecture)是NVIDIA推出的并行计算平台,能够极大地提升GPU加速性能。本文将手把手教你如何在Linux系统上完成配置,确保过程简单明了,即使是初学者也能快速上手。

Linux环境下安装CUDA 11.8全攻略(新手必看的深度学习环境配置指南) Linux安装CUDA 11.8  NVIDIA驱动程序 深度学习环境 CUDA Toolkit 第1张

第一步:检查基础环境

在安装之前,我们需要确保系统拥有合适的NVIDIA驱动程序。请在终端输入以下命令:

nvidia-smi

如果显示了显卡信息且驱动版本支持11.8(通常驱动版本需高于450.80.02),则可以继续。

第二步:下载CUDA Toolkit 11.8

访问NVIDIA官网,选择对应的Linux版本和架构(通常是x86_64)。推荐使用“runfile”方式安装,因为它相对独立且稳定。

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

第三步:执行安装程序

使用管理员权限运行下载好的安装包。注意,在弹出的界面中,如果已经安装了驱动,请勾选取消“Driver”选项,只安装CUDA Toolkit

sudo sh cuda_11.8.0_520.61.05_linux.run

跟随提示输入“accept”并选择“Install”即可。

第四步:配置环境变量

这是至关重要的一步。为了让系统识别CUDA命令,需要编辑 ~/.bashrc 文件:

export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存后执行 source ~/.bashrc 使其生效。完成环境变量配置后,你的系统就能调用CUDA进行计算了。

第五步:验证安装结果

最后,检查CUDA是否正确安装到你的深度学习环境中:

nvcc -V

如果终端输出了“Cuda compilation tools, release 11.8”等字样,恭喜你,安装圆满成功!

SEO关键词回顾:本文详细讲解了 Linux安装CUDA 11.8 的步骤,涵盖了 NVIDIA驱动程序 的检查、 环境变量配置 的细节以及 深度学习环境 的验证,助你轻松开启AI之旅。