GPU or TPU

king12 · 发表于 2022-3-12 08:16:43

星级打分

平均分:NAN 参与人数:0 我的评分:未评

哪位大佬给讲解下，DFL能不能使用tpu进行训练呢？tpu适用于神经网络，DFL理论上也是基于神经网络，进行模型加载计算，是不是可以用TPU训练DFL呢？

dsyrock · 发表于 2022-3-12 08:42:44

如果你说colab上那个的话，tpu完全用不了

忘忧之人 · 发表于 2022-3-12 09:03:08

不懂帮顶

wcw888 · 发表于 2022-3-12 09:04:01

貌似用不了，选择tpu结果还是cpu在运行

3264659958 · 发表于 2022-3-12 22:24:43

提示: 作者被禁止或删除内容自动屏蔽

JACKYMASTER · 发表于 2022-3-12 23:22:17

tpu用的是cpu的算力，gpu用的是显卡的算力。DFL好像没有tpu算力的代码写入，如果DFL开源，其实可以自己试试加入相应代码，应该就能使用tpu，但估计比较复杂。

3264659958 · 发表于 2022-3-14 09:16:07

提示: 作者被禁止或删除内容自动屏蔽

		自动登录	找回密码
密码			立即注册（仅限QQ邮箱）