淘先锋技术网

首页 1 2 3 4 5 6 7

本人是在pycharm运行代码,安装了tensorflow版本2.13.0

先运行代码查看有没有使用GPU加速:

import tensorflow as tf


# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    physical_devices = tf.config.list_physical_devices('GPU')
    print("Num GPUs:", len(physical_devices))

如果运行结果为0,则没有使用GPU加速训练。

根据官网,安装了适用于 macOS GPU 的Metal PluggableDevice,在pycharm的terminal输入命令:

python3 -m pip install tensorflow-metal

安装成功后,再次运行顶部代码:

 

 

运行结果GPU数量是1,安装成功了!