gpt4 book ai didi

tensorflow - 在 TensorFlow 中添加新设备

转载 作者:行者123 更新时间:2023-12-04 11:35:44 26 4
gpt4 key购买 nike

在 TensorFlow 中, session 将图形操作放置到设备上,例如 CPU 或 GPU,并提供执行它们的方法。是否可以向 TensorFlow 添加额外类别的设备?除了CPU和GPU,我们还叫它XPU?

最佳答案

尽管我们没有很好的文档,但可以向 TensorFlow 添加新的设备类型。大约,它需要添加 tensorflow::Device 的新实现。 ,添加 tensorflow::DeviceFactory 的对应实现,并调用以下注册宏:

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);

关于tensorflow - 在 TensorFlow 中添加新设备,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35213172/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com