环境搭建:从零配置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               │   │
│  └──────────────────────────────────────────────┘   │
│                                                     │
└─────────────────────────────────────────────────────┘

推荐配置

组件推荐版本说明
Python3.9 - 3.11AI生态兼容性最好的版本
CondaMiniconda 最新版轻量级,够用即可
IDEVS 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 下载:

  1. 访问 https://www.python.org/downloads/
  2. 下载 Python 3.10 或 3.11
  3. 安装时 务必勾选 “Add Python to PATH”
  4. 验证: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 快捷键(必备)

操作命令模式编辑模式
运行当前cellShift+EnterShift+Enter
新建cell上方A-
新建cell下方B-
删除cellD,D (连按两次)-
切换为代码Y-
切换为MarkdownM-
进入编辑模式-Enter
退出编辑模式Esc-

五、VS Code 配置(推荐)

如果你更喜欢本地编辑器的体验,VS Code + Python 插件是最佳选择。

安装步骤

  1. 下载 VS Code:https://code.visualstudio.com/
  2. 打开 VS Code,按 Ctrl+Shift+X 打开扩展商店
  3. 搜索并安装以下插件:
插件名称用途
Python (Microsoft)Python 语言支持
Jupyter (Microsoft)Notebook 支持
Pylance智能补全和类型检查
Code Runner快速运行代码片段

VS Code 中使用 Jupyter

  1. 创建新文件,保存为 .ipynb 格式
  2. 自动识别为 Notebook,可直接运行 cell
  3. 支持代码补全、调试、可视化

六、核心依赖库安装

现在来安装 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’

检查以下几点:

  1. 是否在正确的虚拟环境中?(conda activate ai_learning)
  2. 是否安装了该模块?(pip list | grep xxx)
  3. IDE 是否使用了正确的解释器?(VS Code 右下角选择)

Q4: GPU 版 PyTorch 安装失败

  1. 确认 NVIDIA 驱动版本(nvidia-smi
  2. 根据 CUDA 版本选择对应的 PyTorch 安装命令
  3. 如果实在不行,先用 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 已安装
  • 测试脚本运行通过
  • 项目目录结构已建立

📊 本篇安装的所有工具

类别工具版本建议
语言Python3.10
环境管理Miniconda最新版
交互式编程JupyterLab4.0+
代码编辑VS Code最新版
数值计算NumPy1.24+
数据分析Pandas2.0+
可视化Matplotlib3.7+
机器学习Scikit-learn1.3+
深度学习PyTorch2.0+
大模型Transformers4.33+

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐