返回介绍

安装

发布于 2025-05-02 13:36:23 字数 1280 浏览 0 评论 0 收藏

作者: Morvan 编辑: Morvan

本节内容包括:

支持的系统

PyTorch 暂时只支持 MacOS, Linux. 暂不支持 Windows! (可怜的 Windows 同学们.. 又被抛弃了). 不过说不定像 Tensorflow 一样,因为 Windows 用户的强烈要求,他们在某天就突然支持了。

安装

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 ).

如果你觉得这篇文章或视频对你的学习很有帮助,请你也分享它,让它能再次帮助到更多的需要学习的人。

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
    我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。