加入收藏 | 设为首页 | 会员中心 | 我要投稿 站长网 (https://www.shuangqin.cn/)- 应用程序、AI行业应用、CDN、低代码、区块链!
当前位置: 首页 > 服务器 > 搭建环境 > Linux > 正文

Linux深度学习环境搭建全流程指南

发布时间:2026-08-27 11:12:37 所属栏目:Linux 来源:DaWei
导读:  选择稳定且社区支持良好的Linux发行版,推荐Ubuntu 20.04/22.04或CentOS Stream 9。安装时勾选“最小化安装”,避免冗余服务占用资源,确保系统更新至最新:sudo apt update && sudo apt upgrade -y(Ubuntu)或

  选择稳定且社区支持良好的Linux发行版,推荐Ubuntu 20.04/22.04或CentOS Stream 9。安装时勾选“最小化安装”,避免冗余服务占用资源,确保系统更新至最新:sudo apt update && sudo apt upgrade -y(Ubuntu)或 sudo dnf update -y(CentOS)。


2026AI模拟图,仅供参考

  验证并安装GPU驱动。NVIDIA显卡用户需先禁用nouveau驱动,再从官网下载对应版本.run文件安装;安装完成后运行nvidia-smi确认驱动正常加载。注意驱动版本需与后续CUDA兼容,可参考NVIDIA官方兼容性矩阵。


  安装CUDA Toolkit与cuDNN。优先使用官方runfile或deb包方式,避免系统级包管理器导致版本冲突。安装后将/usr/local/cuda/bin加入PATH,/usr/local/cuda/lib64加入LD_LIBRARY_PATH,并写入~/.bashrc生效。cuDNN需解压后手动复制头文件与库文件到CUDA目录。


  创建独立Python环境。推荐使用Miniconda而非系统Python,下载并安装后执行conda init bash,重启终端。新建环境:conda create -n dl python=3.9,激活后安装基础工具:pip install --upgrade pip setuptools wheel。


  安装深度学习框架。PyTorch用户执行pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118(根据CUDA版本调整URL);TensorFlow用户使用pip install tensorflow[and-cuda](TF 2.10+)或指定GPU版本的whl包。务必核对框架、CUDA、cuDNN三者版本匹配关系。


  验证安装是否成功。在Python中依次导入torch/tensorflow,检查torch.cuda.is_available()或tf.test.is_built_with_cuda()返回True,并运行简单张量运算确认GPU加速可用。建议保存一份requirements.txt记录精确版本,便于环境复现。


  日常开发中推荐VS Code配以Remote-SSH或Jupyter Lab作为主力工具。通过conda activate dl统一管理依赖,避免pip与conda混用导致环境紊乱。定期清理conda缓存(conda clean --all)和旧内核(jupyter kernelspec list/remove),保持系统整洁高效。

(编辑:站长网)

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

    推荐文章