AI入门第一步:10分钟搭好你的深度学习环境
手把手教你安装Python、CUDA和PyTorch,写一行代码验证环境,零基础也能跑通第一个AI模型。
准备环境:你需要装什么?
在开始跑AI模型之前,先要装好三样东西:Python(一种编程语言,AI代码主要用它写)、CUDA(让显卡加速计算的工具,如果你有NVIDIA显卡才需要)、PyTorch(一个深度学习框架,帮你快速搭建AI模型)。别怕,跟着步骤走,10分钟搞定。
安装步骤:一步步来
- 安装Python:去官网 python.org 下载最新版(比如3.10或3.11),安装时勾选“Add Python to PATH”,然后一路点“下一步”。
- 检查是否装好:打开命令行(Windows按Win+R,输入cmd回车),输入
python --version,看到版本号就对了。 - 安装PyTorch:打开 PyTorch 官网 pytorch.org,点“Get Started”,选择你的系统(Windows/Linux/Mac)和包管理器(一般选pip),复制生成的命令(比如
pip install torch torchvision torchaudio),粘贴到命令行里回车。等几分钟下载完成。 - (可选)安装CUDA:如果你有NVIDIA显卡,去NVIDIA官网下载CUDA Toolkit(建议11.8或12.1),安装时选默认路径。装完后在命令行输入
nvcc --version验证。
验证是否成功:写一行代码跑通
打开命令行,输入 python 进入交互模式,然后复制下面这行代码按回车:
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