加入收藏 | 设为首页 | 会员中心 | 我要投稿 站长网 (https://www.92codes.com/)- 云服务器、云原生、边缘计算、云计算、混合云存储!
当前位置: 首页 > 服务器 > 搭建环境 > Linux > 正文

Linux机器学习环境搭建:数据库配置与快速运行指南

发布时间:2026-08-09 14:16:33 所属栏目:Linux 来源:DaWei
导读:  Linux系统是机器学习开发的主流平台,稳定高效且生态丰富。本文聚焦于数据库配置与环境快速启动,避免冗余步骤,适用于Ubuntu/Debian及CentOS/RHEL系发行版。   数据库选用PostgreSQL,因其支持JSONB、全文检

  Linux系统是机器学习开发的主流平台,稳定高效且生态丰富。本文聚焦于数据库配置与环境快速启动,避免冗余步骤,适用于Ubuntu/Debian及CentOS/RHEL系发行版。


  数据库选用PostgreSQL,因其支持JSONB、全文检索和高并发,适合存储特征元数据、模型版本与实验日志。安装命令简洁:Ubuntu/Debian用户执行sudo apt update && sudo apt install -y postgresql postgresql-contrib;CentOS/RHEL则运行sudo dnf install -y postgresql-server postgresql-contrib,随后初始化数据库集群(postgresql-setup initdb)并启用服务(sudo systemctl enable --now postgresql)。


  为安全与易用,创建专用数据库用户与库:切换至postgres用户(sudo -u postgres psql),执行CREATE DATABASE ml_db;、CREATE USER ml_user WITH PASSWORD 'secure_pass123';、GRANT ALL PRIVILEGES ON DATABASE ml_db TO ml_user;。退出后验证连接:psql -h localhost -U ml_user -d ml_db,输入密码即可成功进入。


  Python环境建议使用Miniconda(轻量、免root),直接下载并安装:wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && bash Miniconda3-latest-Linux-x86_64.sh -b -p $HOME/miniconda3,然后运行$HOME/miniconda3/bin/conda init bash并重新登录终端。激活base环境后,创建专用ML环境:conda create -n ml-env python=3.10,再conda activate ml-env。


AI绘图结果,仅供参考

  核心依赖一键安装:pip install numpy pandas scikit-learn torch torchvision tensorflow psycopg2-binary jupyter notebook。其中psycopg2-binary是Python连接PostgreSQL的驱动,无需编译依赖,适合快速部署。若需GPU支持,PyTorch与TensorFlow会自动适配CUDA(确保已安装NVIDIA驱动及cuDNN)。


  快速运行示例:启动Jupyter Notebook(jupyter notebook --ip=0.0.0.0 --port=8888 --no-browser --allow-root),在浏览器中访问http://localhost:8888(或服务器IP)。新建Notebook,首单元格导入数据库模块:import psycopg2;第二单元格测试连接:conn = psycopg2.connect("host='localhost' dbname='ml_db' user='ml_user' password='secure_pass123'");第三单元格写入简单记录:cursor = conn.cursor(); cursor.execute("CREATE TABLE IF NOT EXISTS experiments (id SERIAL PRIMARY KEY, name TEXT, timestamp TIMESTAMPTZ DEFAULT NOW())"); conn.commit(); cursor.close(); conn.close()——运行即完成基础闭环。


  建议将上述命令整理为setup.sh脚本:包含用户权限校验、PostgreSQL状态检查、conda路径设置与依赖安装。首次运行后,后续只需source ~/.bashrc && conda activate ml-env && jupyter notebook三步即可复用全部环境。所有配置均不修改系统全局Python或数据库默认端口(5432),避免冲突,保障多项目隔离性。

(编辑:站长网)

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

    推荐文章