美文网首页
五、深度学习框架PyTorch的安装

五、深度学习框架PyTorch的安装

作者: 星光下的胖子 | 来源:发表于2020-05-05 14:26 被阅读0次

    官网地址:https://pytorch.org/

    先选择相关配置,会自动生成安装命令行。
    然后打开电脑终端,复制执行命令行即可进行安装操作。

    演示一、Python的Pytorch安装(Mac下)

    环境选择:

    • Mac系统
    • Pip 或 Conda(如果电脑已安装过Conda,这里就可以选择Conda)
    • Python语言
    • None(没有安装CUDA,所以选None,如果你已安装CUDA可选相应的版本)。

    环境选择完以后,下方 Run this Command 处会生成相应的安装命令。

    环境选择

    拷贝命令,打开终端执行安装操作。

    执行安装命令

    安装成功后,执行如下命令进行验证。

    python  #进入python
    import torch  #引入torch
    print (torch.__version__)  #打印torch版本
    exit()  #退出python
    
    查看torch版本

    由上图可见,安装成功,torch版本为 1.5.0

    演示二、C++的Pytorch安装(Mac下)

    环境选择:

    • Mac系统
    • LibTorch
    • C++
    • None(没有安装CUDA,所以选None,如果你已安装CUDA可选相应的版本)。

    环境选择完以后,下方 Run this Command 处会生成相应的安装命令。

    环境选择

    直接通过链接https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.5.0.zip下载.zip文件。

    下载libtorch

    下载后直接解压.zip文件即可。(unzip -u libtorch-macos-1.5.0.zip)

    相关文章

      网友评论

          本文标题:五、深度学习框架PyTorch的安装

          本文链接:https://www.haomeiwen.com/subject/fvmpghtx.html