macM1安装tensorflow和torch教程(python3.8环境)

macM1安装tensorflow和torch教程(python3.8环境)

为搭建机器学习环境,appleM1的环境搭建大概花了一周摸索,使用pip install 报错,发现是因为arm架构的问题。

1.配置conda环境

首先,conda forge提供的miniforge支持M1芯片,下载miniforge3,选择Apple Silicon版本 下载地址:miniforge下载地址
macM1安装tensorflow和torch教程(python3.8环境)_第1张图片
下载文件到本地,安装conda。

为tensorflow搭建Python环境

miniforge3默认是python3.9,我们安装tensorflow需要3.8版本的python,所以我们需要先创建一个conda环境

conda create -—name python38 python=3.8
# 激活环境
conda activate python38

2.下载tensorflow文件

下载地址:tensorflow下载,下载完成后,使用tar进行解压

tar -xvf tensorflow_macos-0.1alpha1.tar 

按步骤安装链接:https://www.ai-buzz.com/18-steps-to-install-tensorflow_macos-on-m1-macbook-2020/
测试
安装完成后,可以在命令行中使用python命令进入交互模式,测试tensorflow是否正确安装。

import tensorflow

如果能正常引用,没有报错,即为安装成功。

3.安装pytorch

安装torch
截至2021-01-27,torch不能通过conda install torch或者pip install torch安装。但是我们有wheel。
下载wheel文件
下载链接:https://github.com/pytorch/pytorch/issues/48145
下载到本地
如果是Python3.8环境

pip install torch-1.8.0a0-cp38-cp38-macosx_11_0_arm64.whl

如果是Python3.9环境

pip install torch-1.8.0a0-cp39-cp39-macosx_11_0_arm64.whl

测试

import torch

参考:Mac m1芯片anaconda上安装tensorflow
pytorch在appleM1上安装

你可能感兴趣的:(appleM1,arm架构,tensorflow,python,anaconda,pytorch,深度学习)