mac tensorflow gpu 安装(arm64 : Apple silicon)

1、安装miniconda 或者 Anaconda3

bash ~/miniconda.sh -b -p $HOME/miniconda
source ~/miniconda/bin/activate
conda install -c apple tensorflow-deps

#Anaconda3 自己百度安装,都行

2、配置 miniconda 环境变量 ,在~/.zshrc 文件中添加如下代码

source ~/miniconda/bin/activate

3、安装 TensorFlow

python -m pip install tensorflow-macos

4、按照 tensorflow-metal plug-in

python -m pip install tensorflow-metal

5、验证 python 跑下面的python 代码

import tensorflow as tf

cifar = tf.keras.datasets.cifar100
(x_train, y_train), (x_test, y_test) = cifar.load_data()
model = tf.keras.applications.ResNet50(
    include_top=True,
    weights=None,
    input_shape=(32, 32, 3),
    classes=100,)

loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
model.compile(optimizer="adam", loss=loss_fn, metrics=["accuracy"])
model.fit(x_train, y_train, epochs=5, batch_size=64)

打开 活动监视器 -> 窗口 -> GPU历史窗口, 看看有红色代表使用了GPU

参考:
https://developer.apple.com/metal/tensorflow-plugin/

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容