「Mac」PyTorch 安装教学

安装步骤

step1 进入官网

    进入 Pytorch 官网:
https://pytorch.org

step2 选择要安装的版本

进入官网后,就能看到 INSTALL PYTORCH 选项:

    我们可以直接安装最新版本的,也可以点击 Previous versions of PyTorch 选择安装之前版本的。

    这里要特别注意下 CUDA 选项,有的 Mac 电脑是没有 CUDA 的。

    这里我选择安装旧版本 V1.51

step3 执行相应的命令

    按照官网所给的命令,执行即可,如:

conda install pytorch==1.5.1 torchvision==0.6.1 -c pytorch

    开始安装:

    安装成功:

step4 测试是否安装成功

    进入 python 环境,输入以下代码测试是否安装 PyTorch 成功。

import torch
torch.__version__

©️2020 CSDN 皮肤主题: 1024 设计师:上身试试 返回首页