pytorch操作基礎（一）

原创于 2019-01-22 15:17:49 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度學習 #PyTorch

PyTorch 专栏收录该内容

7 篇文章

订阅专栏

本文详细介绍如何在MacOS环境下使用pip安装PyTorch及torchvision模块，为深度学习项目搭建基础框架。涵盖PyTorch官网提供的安装指南，以及如何选择适合个人环境的安装版本。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

深度學習（一）

先用PyTorch搭建深度學習的框架

1.搭建

PyTorch 安装起来很简单, 它自家网页上就有很方便的选择方式 (网页升级改版后可能和下图有点不同):

在这里插入图片描述
所以根据你的情况选择适合你的安装方法, 我已自己为例, 我使用的是 MacOS, 想用 pip 安装, 我的 Python 是 3.5 版的, 我没有 GPU 加速, 那我就按上面的选:
然后根据上面的提示, 我只需要在我的 Terminal 当中输入以下指令就好了:

$ pip install http://download.pytorch.org/whl/torch-0.1.11.post5-cp35-cp35m-macosx_10_7_x86_64.whl
$ pip install torchvision

注意, 我安装的是0.1.11版本的 torch, 你需要去他们网站上看是否有新版本的. 安装 PyTorch 会安装两个模块, 一个是 torch, 一个 torchvision, torch 是主模块, 用来搭建神经网络的, torchvision 是辅模块, 有数据库, 还有一些已经训练好的神经网络等着你直接用, 比如 (VGG, AlexNet, ResNet).