Linux机器学习环境搭建全指南
|
Linux是机器学习开发的主流平台,因其稳定性、开源生态和强大终端支持而广受青睐。选择一个较新的发行版(如Ubuntu 22.04/24.04或Debian 12)可减少兼容性问题,建议使用64位系统并确保至少8GB内存和50GB可用磁盘空间。 安装前先更新系统:执行sudo apt update && sudo apt upgrade -y(Ubuntu/Debian)或sudo dnf update -y(Fedora),以获取最新内核与安全补丁。随后安装基础编译工具和依赖:sudo apt install -y build-essential python3-dev python3-pip git curl wget,这些是后续编译CUDA驱动、安装Python包的前提。 推荐使用pyenv管理Python版本,避免系统Python被污染。通过curl安装pyenv:curl https://pyenv.run | bash,然后将三行配置添加至~/.bashrc(或~/.zshrc),运行source ~/.bashrc后即可安装指定版本,例如pyenv install 3.11.9 && pyenv global 3.11.9。验证执行python --version是否返回预期版本。 pip需升级至最新:pip install --upgrade pip。为提升国内下载速度,可配置清华镜像源:pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/。常用科学计算库一键安装:pip install numpy scipy pandas matplotlib scikit-learn,这些构成数据预处理与经典模型训练的核心栈。 深度学习框架推荐PyTorch与TensorFlow。PyTorch安装应优先采用官方提供的GPU版本指令(如pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121),确保匹配本地CUDA版本;若无NVIDIA显卡,则改用CPU版本。TensorFlow可通过pip install tensorflow(CPU)或tensorflow-cpu安装,避免与PyTorch CUDA驱动冲突。 Jupyter Lab提供交互式开发体验,运行pip install jupyterlab后,启动命令为jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root。如需远程访问,配合SSH端口转发或Nginx反向代理,并设置密码:jupyter lab password。 GPU加速需额外步骤:先确认NVIDIA驱动已正确安装(运行nvidia-smi查看),再根据驱动版本选择对应CUDA Toolkit(推荐CUDA 12.x系列)。从NVIDIA官网下载runfile安装包,禁用nouveau驱动后执行安装。安装完成后,将/usr/local/cuda/bin加入PATH,/usr/local/cuda/lib64加入LD_LIBRARY_PATH。 环境校验不可省略:运行Python代码检查关键组件——import torch; print(torch.cuda.is_available())应返回True;import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))可确认GPU识别状态;import numpy as np; print(np.__version__)验证基础库完整性。所有输出无报错即表示核心环境就绪。
AI生成3D模型,仅供参考 使用pip freeze > requirements.txt导出当前环境依赖清单,便于复现或迁移。日常开发中,建议为不同项目创建虚拟环境:python -m venv ml-env && source ml-env/bin/activate,保障项目隔离性与可维护性。至此,一个生产就绪的Linux机器学习环境已稳定可用。 (编辑:开发网_新乡站长网) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |


浙公网安备 33038102330465号