极客前沿

AI入门第一步:10分钟搭好你的深度学习环境

2026-06-01 01:32
DEV Machine Learning
查看原文

手把手教你安装Python、CUDA和PyTorch,写一行代码验证环境,零基础也能跑通第一个AI模型。

准备环境:你需要装什么?

在开始跑AI模型之前,先要装好三样东西:Python(一种编程语言,AI代码主要用它写)、CUDA(让显卡加速计算的工具,如果你有NVIDIA显卡才需要)、PyTorch(一个深度学习框架,帮你快速搭建AI模型)。别怕,跟着步骤走,10分钟搞定。

安装步骤:一步步来

  1. 安装Python:去官网 python.org 下载最新版(比如3.10或3.11),安装时勾选“Add Python to PATH”,然后一路点“下一步”。
  2. 检查是否装好:打开命令行(Windows按Win+R,输入cmd回车),输入 python --version,看到版本号就对了。
  3. 安装PyTorch:打开 PyTorch 官网 pytorch.org,点“Get Started”,选择你的系统(Windows/Linux/Mac)和包管理器(一般选pip),复制生成的命令(比如 pip install torch torchvision torchaudio),粘贴到命令行里回车。等几分钟下载完成。
  4. (可选)安装CUDA:如果你有NVIDIA显卡,去NVIDIA官网下载CUDA Toolkit(建议11.8或12.1),安装时选默认路径。装完后在命令行输入 nvcc --version 验证。

验证是否成功:写一行代码跑通

打开命令行,输入 python 进入交互模式,然后复制下面这行代码按回车:

Tutorial Image
  • import torch; print(torch.__version__); print(torch.cuda.is_available())

如果看到PyTorch版本号(比如2.0.1)和 True(有显卡)或 False(无显卡),说明安装成功!

下一步可以做什么

你已经能跑AI了!可以试试:

  • 用预训练模型做图片分类(搜索“PyTorch 图像分类 入门”)
  • 用Hugging Face下载一个聊天模型(搜索“transformers 库 快速开始”)
  • 跟着官方教程写一个简单的神经网络(pytorch.org/tutorials)

记住:刚开始别纠结原理,先跑起来,再慢慢理解。遇到报错就复制错误信息去搜索引擎查,99%的问题都有答案。

内容来源

DEV Machine Learning

发布时间

2026-06-01 01:32

返回 AI技术