如何在Tensorflow.JS上激活webgpu后端

阿隆·伯格(Alon Burg)

Tensorflow.js的Github上的当前版本包含一个WEBGPU后端,但是运行时,即使启用了WEBGPU的Chrome Canary,tf.getBackend()webgl什至可以使用。运行tf.backend()也似乎并不表明存在WEBGPU。

绿豆

tf您要导入哪个包?WebGPU后端tfjs默认情况下未捆绑-您需要专门导入它,如下所示:

import * as tf from '@tensorflow/tfjs-backend-webgpu';

const init = async () => {
    await tf.ready();

    // Now we can create tensors and run ops.
    tf.matMul(a, b).print(); 
};

init();

请注意,后端是异步的-您必须await tf.ready()在对其进行任何操作之前先进行调用

希望能有所帮助。

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

需要TensorFlow依赖项。如何在Windows上运行TensorFlow

来自分类Dev

如何在TensorFlow变量上获得损耗的梯度?

来自分类Dev

如何在TensorFlow中使用线性激活函数?

来自分类Dev

Docker上的Tensorflow:如何在Jupyter笔记本上保存工作?

来自分类Dev

如何在spyder上使用tensorflow?

来自分类Dev

如何在一个内核上运行Tensorflow?

来自分类Dev

如何在带有TensorFlow后端的Keras中使用其他GPU?

来自分类Dev

如何在Tensorflow上使用文件列表作为Sagemaker上的训练集?

来自分类Dev

如何在Google计算引擎上运行Tensorflow GPU容器?

来自分类Dev

如何在BlueData上使用Tensorflow从datatap读取和写入?

来自分类Dev

如何在Google Cloud Deep Learning VM上安装tensorflow-transform?

来自分类Dev

如何在具有tensorflow2和keras的多GPU上训练模型?

来自分类Dev

我如何在tensorflow.js中定义自己的标签

来自分类Dev

vscode如何在pugjs上激活emmet?

来自分类Dev

如何在具有tensorflow v2.x后端的keras中加载具有tensorflow v1.x后端的keras模型?

来自分类Dev

如何在Google Colab上安装Tensorflow的先前版本?

来自分类Dev

如何在Python 3.7上安装TensorFlow

来自分类Dev

如何在Kaggle上安装最新版本的tensorflow?

来自分类Dev

如何在Kaggle Notebook中的GPU上运行Tensorflow的Keras model.fit()函数?

来自分类Dev

如何在Android设备上使用Node js后端运行React Native应用程序?

来自分类Dev

如何在Windows 10上的python中为TensorFlow设置带有CUDA的GPU?

来自分类Dev

如何在医学数据上训练tensorflow.js

来自分类Dev

如何在Windows 10上为TensorFlow安装TensorRT的Wheel包?

来自分类Dev

如何在Blogger模板上激活JavaScript

来自分类Dev

如何在 Tensorflow 中更改 LSTM 单元中的激活函数

来自分类Dev

如何在 tensorflow-r1.0 中裁剪 grad_and_var 元组上的梯度范数?

来自分类Dev

如何在 Tensorflow 上共享 RNN 的变量

来自分类Dev

如何在 Cityscapes 上训练 TensorFlow 的 deeplab 模型?

来自分类Dev

如何在 ubuntu 16.04 上安装 Tensorflow GPU

Related 相关文章

  1. 1

    需要TensorFlow依赖项。如何在Windows上运行TensorFlow

  2. 2

    如何在TensorFlow变量上获得损耗的梯度?

  3. 3

    如何在TensorFlow中使用线性激活函数?

  4. 4

    Docker上的Tensorflow:如何在Jupyter笔记本上保存工作?

  5. 5

    如何在spyder上使用tensorflow?

  6. 6

    如何在一个内核上运行Tensorflow?

  7. 7

    如何在带有TensorFlow后端的Keras中使用其他GPU?

  8. 8

    如何在Tensorflow上使用文件列表作为Sagemaker上的训练集?

  9. 9

    如何在Google计算引擎上运行Tensorflow GPU容器?

  10. 10

    如何在BlueData上使用Tensorflow从datatap读取和写入?

  11. 11

    如何在Google Cloud Deep Learning VM上安装tensorflow-transform?

  12. 12

    如何在具有tensorflow2和keras的多GPU上训练模型?

  13. 13

    我如何在tensorflow.js中定义自己的标签

  14. 14

    vscode如何在pugjs上激活emmet?

  15. 15

    如何在具有tensorflow v2.x后端的keras中加载具有tensorflow v1.x后端的keras模型?

  16. 16

    如何在Google Colab上安装Tensorflow的先前版本?

  17. 17

    如何在Python 3.7上安装TensorFlow

  18. 18

    如何在Kaggle上安装最新版本的tensorflow?

  19. 19

    如何在Kaggle Notebook中的GPU上运行Tensorflow的Keras model.fit()函数?

  20. 20

    如何在Android设备上使用Node js后端运行React Native应用程序?

  21. 21

    如何在Windows 10上的python中为TensorFlow设置带有CUDA的GPU?

  22. 22

    如何在医学数据上训练tensorflow.js

  23. 23

    如何在Windows 10上为TensorFlow安装TensorRT的Wheel包?

  24. 24

    如何在Blogger模板上激活JavaScript

  25. 25

    如何在 Tensorflow 中更改 LSTM 单元中的激活函数

  26. 26

    如何在 tensorflow-r1.0 中裁剪 grad_and_var 元组上的梯度范数?

  27. 27

    如何在 Tensorflow 上共享 RNN 的变量

  28. 28

    如何在 Cityscapes 上训练 TensorFlow 的 deeplab 模型?

  29. 29

    如何在 ubuntu 16.04 上安装 Tensorflow GPU

热门标签

归档