环境搭建:从零配置AI开发环境
·
环境搭建:从零配置AI开发环境
前言
"工欲善其事,必先利其器。"在开始AI学习之旅前,搭建一个完善的开发环境是第一步,也是最重要的一步。很多初学者在这一步就卡住了——Python版本冲突、库安装失败、环境变量搞不定……
别担心!这篇文章将手把手带你从零开始,搭建一个完整、规范的AI开发环境。跟着操作,30分钟搞定!
一、环境概览
我们要安装什么?
┌─────────────────────────────────────────────────────┐
│ AI 开发环境全景 │
├─────────────────────────────────────────────────────┤
│ │
│ ┌──────────┐ ┌──────────┐ ┌──────────────────┐ │
│ │ Python │ │ Conda │ │ IDE / 编辑器 │ │
│ │ 3.9-3.11 │ │ 虚拟环境 │ │ VS Code │ │
│ └──────────┘ └──────────┘ │ Jupyter Notebook │ │
│ └──────────────────┘ │
│ │
│ ┌──────────────────────────────────────────────┐ │
│ │ 核心依赖库 │ │
│ │ NumPy Pandas Matplotlib Scikit-learn │ │
│ │ PyTorch/TensorFlow HuggingFace │ │
│ └──────────────────────────────────────────────┘ │
│ │
└─────────────────────────────────────────────────────┘
推荐配置
| 组件 | 推荐版本 | 说明 |
|---|---|---|
| Python | 3.9 - 3.11 | AI生态兼容性最好的版本 |
| Conda | Miniconda 最新版 | 轻量级,够用即可 |
| IDE | VS Code + Jupyter | 主力开发工具 |
| 包管理 | pip + conda | 两者配合使用 |
二、Python 安装
方案一:使用 Anaconda/Miniconda(推荐 ⭐)
为什么推荐 Conda?
- 自动管理 Python 版本和依赖
- 虚拟环境隔离,避免冲突
- 一键安装常用科学计算库
- 跨平台支持
步骤 1:下载 Miniconda
访问官网:https://docs.conda.io/en/latest/miniconda.html
根据你的系统选择:
- Windows:
Miniconda3 Windows-x86_64.exe - macOS:
Miniconda3 macOS-arm64.pkg(M1/M2) 或Miniconda3 macOS-x86_64.pkg(Intel) - Linux:
Miniconda3-latest-Linux-x86_64.sh
步骤 2:安装
Windows:
# 双击下载的 .exe 文件
# ✅ 勾选 "Add Miniconda to PATH"
# 点击 Install,等待完成
macOS/Linux:
# 给予执行权限并运行
chmod +x Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh
# 按照提示操作,最后输入 yes 初始化
步骤 3:验证安装
打开终端(Terminal / CMD / PowerShell):
# 检查 conda 版本
conda --version
# 预期输出类似:conda 23.x.x
💡 小技巧:如果提示
conda 不是内部或外部命令,需要重启终端或手动添加环境变量。
方案二:直接安装 Python(备选)
如果不想用 Conda,可以直接从 python.org 下载:
- 访问 https://www.python.org/downloads/
- 下载 Python 3.10 或 3.11
- 安装时 务必勾选 “Add Python to PATH”
- 验证:
python --version
三、虚拟环境管理
为什么需要虚拟环境?
想象一下这个场景:
- 项目A需要
numpy==1.21 - 项目B需要
numpy==1.24 - 如果装在同一个环境,就会冲突 ❌
虚拟环境就是给每个项目一个独立的"房间",互不干扰 ✅
创建虚拟环境
# 创建名为 ai_learning 的环境,指定 Python 版本
conda create -n ai_learning python=3.10 -y
# 激活环境
# Windows:
conda activate ai_learning
# macOS/Linux:
source activate ai_learning
# 或
conda activate ai_learning
📌 提示:激活后,终端前面会出现
(ai_learning)标识,说明已进入虚拟环境。
常用 conda 命令速查
| 命令 | 作用 |
|---|---|
conda env list | 查看所有环境 |
conda activate <name> | 激活环境 |
conda deactivate | 退出当前环境 |
conda remove -n <name> --all | 删除环境 |
conda info --envs | 环境详细信息 |
四、Jupyter Notebook 安装与配置
Jupyter 是 AI 学习的神器——交互式编程,写一行看一行结果,非常适合实验和探索。
安装 Jupyter
# 在虚拟环境中执行
pip install jupyterlab
# 或者安装经典版
pip install notebook
启动 Jupyter
# 启动 Jupyter Lab(推荐,功能更强大)
jupyter lab
# 或启动经典版
jupyter notebook
执行后会自动打开浏览器,显示文件管理界面。
Jupyter 快捷键(必备)
| 操作 | 命令模式 | 编辑模式 |
|---|---|---|
| 运行当前cell | Shift+Enter | Shift+Enter |
| 新建cell上方 | A | - |
| 新建cell下方 | B | - |
| 删除cell | D,D (连按两次) | - |
| 切换为代码 | Y | - |
| 切换为Markdown | M | - |
| 进入编辑模式 | - | Enter |
| 退出编辑模式 | Esc | - |
五、VS Code 配置(推荐)
如果你更喜欢本地编辑器的体验,VS Code + Python 插件是最佳选择。
安装步骤
- 下载 VS Code:https://code.visualstudio.com/
- 打开 VS Code,按
Ctrl+Shift+X打开扩展商店 - 搜索并安装以下插件:
| 插件名称 | 用途 |
|---|---|
| Python (Microsoft) | Python 语言支持 |
| Jupyter (Microsoft) | Notebook 支持 |
| Pylance | 智能补全和类型检查 |
| Code Runner | 快速运行代码片段 |
VS Code 中使用 Jupyter
- 创建新文件,保存为
.ipynb格式 - 自动识别为 Notebook,可直接运行 cell
- 支持代码补全、调试、可视化
六、核心依赖库安装
现在来安装 AI 开发最常用的库。
基础数据处理三剑客
# 数值计算基础
pip install numpy==1.24.3
# 数据分析神器
pip install pandas==2.0.3
# 可视化绘图
pip install matplotlib==3.7.2
pip install seaborn==0.12.2
机器学习库
# 传统机器学习算法大全
pip install scikit-learn==1.3.0
# 梯度提升树(竞赛常用)
pip install xgboost==1.7.6
pip install lightgbm==4.0.0
深度学习框架(二选一)
# PyTorch(推荐,研究首选)
# 访问 https://pytorch.org/ 获取适合你系统的安装命令
# CUDA 11.8 示例(NVIDIA GPU):
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# CPU 版本(无 GPU 时用):
pip install torch torchvision torchaudio
# TensorFlow(备选)
pip install tensorflow==2.13.0
大模型相关
# Hugging Face Transformers(预训练模型宝库)
pip install transformers==4.33.2
# LangChain(LLM 应用开发框架)
pip install langchain==0.0.299
# OpenAI API 客户端
pip install openai==0.28.1
其他实用工具
# 进度条显示
pip install tqdm==4.65.0
# 数据集加载和管理
pip install datasets==2.14.4
# 模型轻量级部署
pip install onnxruntime==1.15.1
# API 框架(后续部署用)
pip install fastapi==103.2.0
pip install uvicorn==0.23.2
一键安装所有依赖
我整理了一个完整的 requirements.txt,你可以一键安装:
# requirements.txt
# 数据处理
numpy==1.24.3
pandas==2.0.3
matplotlib==3.7.2
seaborn==0.12.2
# 机器学习
scikit-learn==1.3.0
xgboost==1.7.6
lightgbm==4.0.0
# 深度学习
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0
# 大模型
transformers==4.33.2
langchain==0.0.299
openai==0.28.1
# 工具库
tqdm==4.65.0
datasets==2.14.4
onnxruntime==1.15.1
fastapi==103.2.0
uvicorn==0.23.2
jupyterlab==4.0.5
ipykernel==6.25.0
一键安装命令:
pip install -r requirements.txt
七、验证环境是否正确
创建一个测试文件 test_env.py,运行以下代码:
"""
AI 开发环境测试脚本
运行此脚本验证所有组件是否正常安装
"""
print("=" * 50)
print("AI 开发环境验证")
print("=" * 50)
# 1. 测试 Python
import sys
print(f"\n✅ Python 版本: {sys.version}")
# 2. 测试 NumPy
import numpy as np
print(f"✅ NumPy 版本: {np.__version__}")
arr = np.array([1, 2, 3, 4, 5])
print(f" 测试数组: {arr}, 均值: {np.mean(arr)}")
# 3. 测试 Pandas
import pandas as pd
print(f"✅ Pandas 版本: {pd.__version__}")
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
print(f" 测试 DataFrame:\n{df}")
# 4. 测试 Matplotlib
import matplotlib
print(f"✅ Matplotlib 版本: {matplotlib.__version__}")
# 5. 测试 Scikit-learn
import sklearn
print(f"✅ Scikit-learn 版本: {sklearn.__version__}")
from sklearn.linear_model import LinearRegression
model = LinearRegression()
print(f" 线性回归模型创建成功")
# 6. 测试 PyTorch(可能较慢)
try:
import torch
print(f"✅ PyTorch 版本: {torch.__version__}")
print(f" CUDA 可用: {torch.cuda.is_available()}")
if torch.cuda.is_available():
print(f" GPU 设备: {torch.cuda.get_device_name(0)}")
except ImportError:
print("⚠️ PyTorch 未安装")
# 7. 测试 Transformers
try:
import transformers
print(f"✅ Transformers 版本: {transformers.__version__}")
except ImportError:
print("⚠️ Transformers 未安装")
print("\n" + "=" * 50)
print("🎉 环境验证完成!")
print("=" * 50)
运行方式:
python test_env.py
预期输出示例:
==================================================
AI 开发环境验证
==================================================
✅ Python 版本: 3.10.11 (main, Apr 5 2023, ...)
✅ NumPy 版本: 1.24.3
测试数组: [1 2 3 4 5], 均值: 3.0
✅ Pandas 版本: 2.0.3
测试 DataFrame:
A B
0 1 4
1 2 5
2 3 6
✅ Matplotlib 版本: 3.7.2
✅ Scikit-learn 版本: 1.3.0
线性回归模型创建成功
✅ PyTorch 版本: 2.0.1
CUDA 可用: True
GPU 设备: NVIDIA GeForce RTX 3080
✅ Transformers 版本: 4.33.2
==================================================
🎉 环境验证完成!
==================================================
八、常见问题排查
Q1: pip 安装很慢怎么办?
使用国内镜像源:
# 临时使用
pip install 包名 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 永久设置
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
其他镜像源:
- 阿里云:
https://mirrors.aliyun.com/pypi/simple/ - 中科大:
https://pypi.mirrors.ustc.edu.cn/simple/ - 豆瓣:
https://pypi.douban.com/simple/
Q2: Conda 下载慢?
配置镜像源:
# 清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
Q3: ModuleNotFoundError: No module named ‘xxx’
检查以下几点:
- 是否在正确的虚拟环境中?(
conda activate ai_learning) - 是否安装了该模块?(
pip list | grep xxx) - IDE 是否使用了正确的解释器?(VS Code 右下角选择)
Q4: GPU 版 PyTorch 安装失败
- 确认 NVIDIA 驱动版本(
nvidia-smi) - 根据 CUDA 版本选择对应的 PyTorch 安装命令
- 如果实在不行,先用 CPU 版本学习,不影响入门
Q5: Jupyter 无法自动打开浏览器
手动复制终端输出的 URL 到浏览器地址栏。
九、项目目录结构推荐
良好的目录习惯能让你的学习事半功倍:
ai_learning/ # 项目根目录
├── notebooks/ # Jupyter Notebook 文件
│ ├── 01_data_exploration.ipynb
│ ├── 02_first_model.ipynb
│ └── ...
├── src/ # Python 模块代码
│ ├── data_processing.py
│ ├── model_training.py
│ └── utils.py
├── data/ # 数据集
│ ├── raw/ # 原始数据
│ └── processed/ # 处理后的数据
├── models/ # 保存的模型
├── outputs/ # 输出结果(图表等)
├── requirements.txt # 依赖清单
├── README.md # 项目说明
└── .gitignore # Git 忽略规则
十、总结与检查清单
✅ 环境搭建检查清单
- Python 3.9-3.11 已安装
- Conda/Miniconda 已安装并可使用
- 虚拟环境
ai_learning已创建并激活 - Jupyter Notebook/Lab 可正常启动
- VS Code 及 Python 插件已安装(可选)
- NumPy、Pandas、Matplotlib 已安装
- Scikit-learn 已安装
- PyTorch 或 TensorFlow 已安装
- 测试脚本运行通过
- 项目目录结构已建立
📊 本篇安装的所有工具
| 类别 | 工具 | 版本建议 |
|---|---|---|
| 语言 | Python | 3.10 |
| 环境管理 | Miniconda | 最新版 |
| 交互式编程 | JupyterLab | 4.0+ |
| 代码编辑 | VS Code | 最新版 |
| 数值计算 | NumPy | 1.24+ |
| 数据分析 | Pandas | 2.0+ |
| 可视化 | Matplotlib | 3.7+ |
| 机器学习 | Scikit-learn | 1.3+ |
| 深度学习 | PyTorch | 2.0+ |
| 大模型 | Transformers | 4.33+ |
更多推荐


所有评论(0)