Mac m1 安装tensorflow

文章目录

  • 一、安装 miniconda
  • 二、 tensorflow-apple-metal.yml
  • 三、虚拟环境
    • 设置为默认源
    • 创建虚拟环境
    • 激活
    • 注册
    • 测试
  • 总结


一、安装 miniconda

从官网安装最新版本 传送门
安装后,旧版本的会被卸载,之前的虚拟环境仍然保留,但不能activate,需要把它们添加进来。

conda info --envs # 查看虚拟环境
conda config --append envs_dirs /Users/admin/opt/miniconda3/envs # 此处是你之前的虚拟环境所在的上一层目录(envs)
# 好了,这一步就完成了。

二、 tensorflow-apple-metal.yml

创建该文件,并保存。
内容如下:

name: tensorflow
channels:
  - apple
  - conda-forge
dependencies:
    - python=3.10
    - pip>=19.0
    - jupyter
    - scikit-learn
    - scipy
    - pandas
    - pandas-datareader
    - matplotlib
    - pillow
    - tqdm
    - requests
    - h5py
    - pyyaml
    - flask
    - boto3
    - ipykernel
    - pip:
        - tensorflow-macos
        - tensorflow-metal
        - bayesian-optimization
        - gym
        - kaggle

三、虚拟环境

命令如下:
注意,使用pip下载依赖的话,国内源是没有该插件的,所以需要使用默认源 即(https://pypi.org/simple/),最好使用梯子

设置为默认源

pip config set global.index-url https://pypi.org/simple/

创建虚拟环境

conda env create -f tensorflow-apple-metal.yml -n tensorflow

激活

conda activate tensorflow

注册

python -m ipykernel install --user --name tensorflow --display-name "Python 3.10 (tensorflow)"

测试

import sys

import tensorflow.keras
import pandas as pd
import sklearn as sk
import tensorflow as tf
import platform

print(f"Python Platform: {platform.platform()}")
print(f"Tensor Flow Version: {tf.__version__}")
print(f"Keras Version: {tensorflow.keras.__version__}")
print()
print(f"Python {sys.version}")
print(f"Pandas {pd.__version__}")
print(f"Scikit-Learn {sk.__version__}")
gpu = len(tf.config.list_physical_devices('GPU'))>0
print("GPU is", "available" if gpu else "NOT AVAILABLE")

总结

在Youtube上看到的,原文章如下,祝各位成功!

原教程: https://github.com/jeffheaton/t81_558_deep_learning/blob/master/install/tensorflow-install-mac-metal-jan-2023.ipynb

你可能感兴趣的:(tensorflow,python,macos,tensorflow)