MacBook Pro运行深度学习代码——Tensorflow_macos

电脑配置

先说一下电脑的配置,本人还是非常想尝鲜M1芯片机器的测试,但是只有MacBook Pro16’的机器,只怪上半年买的机器,下半年就上新M1,性能还是质的飞跃,生气(看窗外风景30分钟后~)
MacBook Pro运行深度学习代码——Tensorflow_macos_第1张图片
还有一张不知道效果咋样的显——AMD Radeon Pro 5300M
MacBook Pro运行深度学习代码——Tensorflow_macos_第2张图片

安装环境——Tensorflow_macos

tensorflow_macos 基于tensorflow2.4版本
github主页
tensorflow_macos下载

有两种方式下载安装:

  1. 官方给出的快捷方式 /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/apple/tensorflow_macos/master/scripts/download_and_install.sh)"
  2. 就是下载解压,在运行命令安装

下载安装

下载解压后的文件目录
MacBook Pro运行深度学习代码——Tensorflow_macos_第3张图片
运行脚本安装

查看脚本中的命令: sudo /bin/bash ~/Downloads/tensorflow_macos/install_venv.sh --help

本人的选择:sudo /bin/bash ~/Downloads/tensorflow_macos/install_venv.sh -p
运行-p命令后会让你输入安装目录,输入目录后知道安装完成即可。

运行代码

运行代码前需要引入安装目录,作为依赖。pycharm设置:

如果你遇到了文件无法选择,说明你的权限不够,我这里直接暴力解决:chmod -R 777 ENV_PATH

设置GPU运行():

  1. from tensorflow.python.compiler.mlcompute import mlcompute
  2. mlcompute.set_mlc_device(device_name='gpu') 默认参数为any

等待pycharm扫描文件结束后就可以运行自己的代码了!!!是不是很开心,我到这一步的时候很是兴奋~

测试对比

GPU运行结果
GPU运行结果
CPU运行结果(版本tensorflow2.3)
CPU运行结果
运行时间差距还是相当大的。

GPU训练时的监控

在macbook上训练神经网络也只是勉勉强强,不知道以后会不会好,看来电脑也是要换了。看窗外风景去了~

你可能感兴趣的:(人工智能,tensorflow,深度学习,macos,macbook)