极客前沿

AI入门:零基础搭建你的第一个AI环境

2026-05-23 01:31
DEV Tutorial
查看原文

手把手教你安装Python和PyTorch,跑通第一个AI模型,避开新手常见坑。

很多人觉得AI开发门槛高,其实从零开始并不难。这篇文章带你装好环境,跑通第一个AI程序,全程不超过30分钟。

准备环境:装好Python和编辑器

AI开发最常用的语言是Python(一种简单易学的编程语言,就像写菜谱一样告诉电脑做什么)。你需要先安装它。

  • 打开浏览器,访问 python.org,点击“Downloads”下载最新版(建议3.9或以上)。
  • 安装时记得勾选“Add Python to PATH”(这样电脑才知道你在哪里输入python命令)。
  • 装完后,打开命令行(Windows搜索“cmd”,Mac打开“终端”),输入 python --version,如果显示版本号就成功了一半。
  • 再装一个编辑器:推荐 VS Code(免费好用的代码记事本),去code.visualstudio.com下载安装。

常见坑: 安装Python时忘记勾选“Add to PATH”,之后运行python命令会报错。如果发生,卸载重装并勾选即可。

安装AI框架:PyTorch

AI框架就像做菜的锅铲——帮你省去重复造轮子。我们用PyTorch(一个流行的AI工具包,能快速搭建神经网络)来做第一个实验。

Tutorial Image
  1. 打开命令行(cmd或终端)。
  2. 输入以下命令并按回车:
    pip install torch torchvision torchaudio
    耐心等待下载安装(大概5-10分钟)。
  3. 如果提示“pip不是内部命令”,说明Python没装好,回头检查PATH。

常见坑: 网络慢导致下载失败。可以换个镜像源,比如国内用户用:pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

验证是否成功:跑通第一个AI程序

我们写一行代码来确认PyTorch能正常工作。

  1. 打开VS Code,新建文件,命名为 test.py
  2. 输入以下内容:
    import torch
    print(torch.__version__)
    print('CUDA available:', torch.cuda.is_available())
  3. 按Ctrl+S保存,然后点击右上角的运行按钮(或按F5)。
  4. 如果输出类似 1.13.0CUDA available: False,说明安装成功!

注意:CUDA是让显卡加速的,没有显卡也能运行,就是慢一点。

下一步可以做什么

你已经成功搭建了AI环境!接下来可以:

  • 尝试用PyTorch训练一个简单的数字识别模型(网上搜“PyTorch MNIST教程”)。
  • 或者学习用预训练模型做图片分类,比如用ResNet(一个已经训练好的图像识别模型)识别猫狗。

记住:先跑通,再理解。遇到问题用百度或Stack Overflow搜报错信息,大部分坑都有人踩过。

内容来源

DEV Tutorial

发布时间

2026-05-23 01:31

返回 AI技术