第3篇:AI开发环境搭建:Anaconda + Jupyter + VS Code(含CUDA配置)
本文详细介绍了如何搭建专业AI开发环境,包含5个关键步骤:1)安装Anaconda并验证;2)使用Conda管理虚拟环境;3)配置Jupyter Notebook;4)设置VSCode编辑器;5)配置GPU加速(CUDA+PyTorch)。重点解决了环境隔离、依赖管理、GPU支持等常见问题,提供了环境导出、镜像源配置等实用技巧,并附有常见问题解决方案。最后预告了后续将讲解的AI数学基础内容。
摘要:
本文手把手教你搭建专业的AI开发环境,涵盖Anaconda虚拟环境管理、Jupyter Notebook高效使用、VS Code代码编辑器配置,并详细讲解如何解决CUDA、cuDNN、PyTorch GPU支持等常见问题。适合AI初学者、学生和转行开发者,确保你在第一步就少走弯路。
一、为什么需要专业的AI开发环境?
在开始写AI代码前,我们必须先解决“在哪里写”和“依赖怎么管”的问题。
1.1 常见痛点
- ❌ “我装了库,但别人运行不了” → 缺乏环境隔离
- ❌ “版本冲突,PyTorch和TensorFlow打架” → 依赖管理混乱
- ❌ “Jupyter打不开” → 路径或内核问题
- ❌ “GPU不工作” → CUDA驱动不匹配
1.2 推荐工具链
工具 | 作用 |
---|---|
Anaconda | Python发行版,集成包管理(conda)和环境管理 |
Jupyter Notebook | 交互式编程环境,适合数据探索与教学 |
VS Code | 轻量级但功能强大的代码编辑器,支持调试与Git |
CUDA | NVIDIA GPU并行计算平台,加速深度学习训练 |
✅ 本篇目标:搭建一个稳定、隔离、可复现的AI开发环境。
二、Step 1:安装Anaconda(推荐Windows/Mac)
2.1 下载与安装
- 访问官网:Download Anaconda Distribution | Anaconda
- 选择对应系统版本(Windows 64-bit / macOS Intel/Apple Silicon)
- 下载后双击安装,建议勾选“Add Anaconda to PATH”(可选,但方便命令行使用)
💡 提示:安装路径避免中文和空格,如
C:\Anaconda3
。
2.2 验证安装
打开 Anaconda Prompt(Windows)或 Terminal(Mac/Linux),输入:
conda --version
python --version
应输出类似:
conda 24.1.2
Python 3.9.18
三、Step 2:使用Conda管理虚拟环境(关键!)
3.1 为什么用虚拟环境?
- 隔离项目依赖(项目A用PyTorch 1.13,项目B用2.0)
- 避免“污染”基础环境
- 方便团队协作(导出环境配置)
3.2 常用Conda命令
# 1. 查看所有环境
conda env list
# 2. 创建新环境(指定Python版本)
conda create -n ai_env python=3.9
# 3. 激活环境
conda activate ai_env
# 4. 退出环境
conda deactivate
# 5. 删除环境
conda env remove -n ai_env
3.3 在环境中安装常用包
激活环境后:
conda activate ai_env
# 安装核心AI库
conda install numpy pandas matplotlib jupyter
# 或使用pip(conda不支持时)
pip install scikit-learn tensorflow torch
✅ 最佳实践:优先使用
conda install
,其次pip
。
四、Step 3:配置Jupyter Notebook
4.1 启动Jupyter
在激活的环境中运行:
jupyter notebook
浏览器将自动打开:http://localhost:8888
4.2 创建并运行Notebook
- 点击 New → Python 3 (ai_env)(确保内核是你的环境)
- 输入代码并按
Shift+Enter
运行
import numpy as np
print("Hello AI World!")
4.3 解决“内核不显示”问题
如果Jupyter中看不到 ai_env
内核:
# 在 ai_env 环境中执行
conda activate ai_env
pip install ipykernel
python -m ipykernel install --user --name ai_env --display-name "Python (ai_env)"
重启Jupyter即可看到新内核。
五、Step 4:配置VS Code(推荐代码编辑器)
5.1 下载与安装
- 访问:https://code.visualstudio.com/
- 安装后,推荐安装以下扩展:
- Python(Microsoft官方)
- Jupyter(支持.ipynb文件)
- Pylance(智能补全)
- GitLens(Git增强)
5.2 配置Python解释器
- 打开VS Code,按
Ctrl+Shift+P
打开命令面板 - 输入
Python: Select Interpreter
- 选择你创建的环境,如:
Python 3.9.18 ('ai_env': conda)
5.3 运行Python代码
创建 hello.py
:
import pandas as pd
print("VS Code + AI 环境配置成功!")
右键选择“在终端中运行Python文件”。
六、Step 5:GPU加速配置(CUDA + cuDNN)
⚠️ 仅适用于NVIDIA显卡用户
6.1 检查GPU支持
nvidia-smi
查看驱动版本和GPU型号。输出应类似:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 535.98 Driver Version: 535.98 CUDA Version: 12.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================================|
| 0 NVIDIA RTX 3060 45C P8 10W / 170W | 1234MiB / 12288MiB | 5% Default |
+-------------------------------+----------------------+----------------------+
6.2 安装CUDA Toolkit(可选)
大多数深度学习框架(如PyTorch)会自带CUDA,无需单独安装。
但若需手动安装:
- 访问:CUDA Toolkit 13.0 Downloads | NVIDIA Developer
- 选择对应系统和版本(与
nvidia-smi
显示的CUDA版本兼容)
6.3 安装PyTorch(自动包含CUDA)
访问:Get Started
选择命令,如:
# Conda安装(推荐)
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
# 或 Pip安装
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
6.4 验证GPU是否可用
在Python中运行:
import torch
print(f"PyTorch版本: {torch.__version__}")
print(f"CUDA可用: {torch.cuda.is_available()}")
print(f"GPU数量: {torch.cuda.device_count()}")
if torch.cuda.is_available():
print(f"当前GPU: {torch.cuda.get_device_name(0)}")
期望输出:
PyTorch版本: 2.3.0+cu121
CUDA可用: True
GPU数量: 1
当前GPU: NVIDIA GeForce RTX 3060
✅ 成功!你的PyTorch已支持GPU加速。
七、常见问题与解决方案(FAQ)
Q1:conda
命令提示“不是内部或外部命令”?
- 原因:Anaconda未添加到PATH。
- 解决:
- 重新安装Anaconda,勾选“Add to PATH”。
- 或使用“Anaconda Prompt”代替系统CMD。
Q2:Jupyter启动报错“port already in use”?
- 解决:更换端口
jupyter notebook --port=8889
Q3:pip install
太慢?
- 解决:使用国内镜像源
或配置全局源:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 包名
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
Q4:PyTorch无法使用GPU?
- 检查步骤:
nvidia-smi
是否正常?- PyTorch安装命令是否包含
cu121
或cuda
? - 驱动版本是否过旧?(建议 ≥ 535)
Q5:如何导出环境以便分享?
# 导出环境配置
conda activate ai_env
conda env export > environment.yml
# 他人可一键创建相同环境
conda env create -f environment.yml
八、总结与最佳实践
本文我们完成了:
- ✅ 安装Anaconda并创建虚拟环境
ai_env
- ✅ 配置Jupyter Notebook并解决内核问题
- ✅ 设置VS Code为AI开发主力编辑器
- ✅ 验证并启用GPU加速(CUDA)
- ✅ 解决了5大常见环境问题
📌 环境管理黄金法则:
- 每个项目一个环境
- 优先使用
conda
安装- 定期导出
environment.yml
九、下一篇文章预告
第4篇:线性代数在AI中的应用:向量、矩阵与张量
我们将深入讲解AI背后的数学基础,包括:
- 向量运算与几何意义
- 矩阵乘法与逆矩阵
- 特征值与特征向量
- 张量在深度学习中的角色
- 所有概念均配NumPy代码实现
数学不再抽象,代码让理论“活”起来!
参考文献
- Anaconda官方文档: Anaconda Documentation - Anaconda
- Jupyter官方指南: Redirecting…
- VS Code Python教程: https://code.visualstudio.com/docs/python/python-tutorial
- PyTorch官方安装指南: Get Started
- 清华大学开源软件镜像站: 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
更多推荐
所有评论(0)