「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__
版权声明:本文为晨旭OvO原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/a1228136188/article/details/109323277