加入收藏 | 设为首页 | 会员中心 | 我要投稿 站长网 (https://www.haochuanmei.com/)- 区块链、物联平台、物联安全、数据迁移、5G!
当前位置: 首页 > 服务器 > 搭建环境 > Linux > 正文

Linux深度学习环境搭建全攻略

发布时间:2026-07-17 14:42:40 所属栏目:Linux 来源:DaWei
导读:  在开始深度学习之前,搭建一个稳定高效的Linux环境是关键第一步。推荐使用Ubuntu 20.04或22.04作为系统基础,因其社区支持完善且兼容性佳。安装时建议选择“带有Ubuntu桌面”的版本,便于后续图形化操作与调试。

  在开始深度学习之前,搭建一个稳定高效的Linux环境是关键第一步。推荐使用Ubuntu 20.04或22.04作为系统基础,因其社区支持完善且兼容性佳。安装时建议选择“带有Ubuntu桌面”的版本,便于后续图形化操作与调试。


2026AI分析图,仅供参考

  完成系统安装后,第一时间更新系统软件包。打开终端输入 sudo apt update && sudo apt upgrade,确保所有组件处于最新状态。同时安装常用开发工具:sudo apt install git vim build-essential,为后续代码管理与编译提供支持。


  接下来是显卡驱动的配置。若使用NVIDIA GPU,需安装官方驱动。通过 Ubuntu 的“附加驱动”工具可自动识别并安装推荐驱动,也可手动执行 sudo ubuntu-drivers autoinstall。安装完成后重启系统,验证驱动是否生效:nvidia-smi 命令应显示显卡信息与驱动版本。


  CUDA 是深度学习的核心计算平台。前往 NVIDIA 官方网站下载对应版本的 CUDA Toolkit(建议选择与你的 GPU 兼容的版本,如 11.8)。通过 .deb 包安装:sudo dpkg -i cuda-repo-.deb,然后执行 sudo apt update && sudo apt install cuda-toolkit-11-8。安装完成后,将 CUDA 路径加入环境变量:在 ~/.bashrc 中添加 export PATH=/usr/local/cuda-11.8/bin:$PATH,再运行 source ~/.bashrc。


  接着是深度学习框架的安装。以 PyTorch 为例,访问 PyTorch 官网获取命令行安装指令。推荐使用 conda 管理环境:创建虚拟环境 conda create -n dl_env python=3.9,激活后执行官方提供的 pip 安装命令。例如:pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118。这将自动安装支持 CUDA 的版本。


  为了提升开发体验,可安装 Jupyter Notebook:conda install jupyterlab。启动后可通过浏览器访问本地开发界面,方便编写和运行代码。同时,推荐使用 VS Code 配合 Python 扩展进行代码编辑,其智能提示与调试功能对模型开发大有帮助。


  检查整个环境是否正常工作。创建一个简单的测试脚本,导入 torch 并检测 GPU 是否可用:print(torch.cuda.is_available())。若返回 True,说明深度学习环境已成功搭建。此时可安心投入模型训练与实验,开启你的深度学习之旅。

(编辑:站长网)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章