网页 综合 更新时间 2026-11-13 22:01:48

PyTorch 安装 | 菜鸟教程

自动蜘蛛 www.runoob.com 正文约 576 字
PyTorch 是一个流行的深度学习框架,支持 CPU 和 GPU 计算。
使用 Anaconda/Miniconda:
PyTorch 官方提供了几种安装方法,可以通过 pip 或 conda 进行安装。
如果你使用 Anaconda 或 Miniconda 管理 Python 环境,使用 conda 安装 PyTorch 可能会更加简单和高效。
如果不了解Anaconda,可以参考: Anaconda 教程
访问 PyTorch 的官方网站 https://pytorch.org/get-started/locally/,网站提供了一个方便的工具,可以根据你的系统配置(操作系统、包管理器、Python版本以及CUDA版本)推荐安装命令。
如果你需要从源代码安装PyTorch,或者想要尝试最新的开发版本,你可以使用以下命令:
这将从 GitHub 克隆 PyTorch 的源代码,并使用 setup.py 进行安装。
Apple Silicon (M1/M2) Mac:
为了确保 PyTorch 已正确安装,我们可以通过执行以下 PyTorch 代码来验证是否安装成功:
如果 torch.cuda.is_available() 输出 True,则说明 PyTorch 成功识别到你的 GPU。
一个简单的实例,构建一个随机初始化的张量:
访问原文 返回搜索 https://www.runoob.com/pytorch/pytorch-install.html
本页为82搜索生成的网页快照,内容来自站内索引;演示数据的原文链接仅作示例。