软件配置 | mac M1安装pytorch踩坑不完全记录

本文记录在mac上安装pytorch的过程。需注意地是,pytorch截至目前为止还不支持M1 GPU。
Last Modified: 2022/1/13


软件配置 | mac M1安装pytorch踩坑不完全记录

  • pytorch
    • 安装
    • 验证
  • 参考链接
    • pytorch
      • 安装
      • 验证


pytorch

因为brew和miniforge3我已经提前安装过,所以这里不再记录了。没有安装过的同志也可以参考这里

安装

  • 基于python3.8创建一个名为pytorch_env的新虚拟环境
    conda create --name pytorch_env python=3.8
  • 激活上面创建的虚拟环境
    conda activate pytorch_env
    在这里插入图片描述
  • 在anaconda官网找到pytorch, 尝试``conda install pytorch torchvision -c pytorch```
    会如下所示的错误…
    软件配置 | mac M1安装pytorch踩坑不完全记录_第1张图片
  • 在pytorch官网看到可以通过pip来安装pytorch
    pip3 install torch torchvision
    软件配置 | mac M1安装pytorch踩坑不完全记录_第2张图片

验证

根据这里,验证是否pytorch已经安装成功。
验证结果呢,如下所示:
软件配置 | mac M1安装pytorch踩坑不完全记录_第3张图片
从结果看是CUDA存在问题。

参考这里,可以查到
软件配置 | mac M1安装pytorch踩坑不完全记录_第4张图片
但其实如果是m1芯片的话,在pytorch的官方论坛上有一个帖子里工程师说:
软件配置 | mac M1安装pytorch踩坑不完全记录_第5张图片
并且在pytorch的github主页可以看到,暂时还不支持在m1 gpu上跑pytorch…
软件配置 | mac M1安装pytorch踩坑不完全记录_第6张图片
所以先看看能不能跑cpu吧。
结果如下所示:
软件配置 | mac M1安装pytorch踩坑不完全记录_第7张图片
嗯,至此安装过程结束。

【持续关注什么时候pytorch支持m1 GPU…


参考链接

写本文时有参考以下链接:

pytorch

安装

M1 mac安装PyTorch的完整步骤指南
Mac怎么查看Homebrew是否安装
conda activate激活环境出错的解决办法
GET STARTED
Apple MacBook M1 芯片 Anaconda安装 Tensorflow Pytorch

验证

(Ubuntu/Window)pytorch验证(pytorch是否安装成功)
Pytorch错误:Torch not compiled with CUDA enabled
How to run on gpu in apple M1
GPU acceleration for Apple’s M1 chip?
苹果 M1 芯片的神经单元可否用于训练 Pytorch 深度学习网络模型?

你可能感兴趣的:(软件配置,pytorch,macos,深度学习)