Sai Raghuram Kaligotla
Sai Raghuram Kaligotla

Reputation: 169

Does TensorFlow.js use the GPU for computations?

The official documentation of TensorFlow.js mentions that it uses WebGL for accessing the GPU. Is WebGL only used for rendering or for computations also?

Upvotes: 6

Views: 351

Answers (1)

Stewart_R
Stewart_R

Reputation: 14485

Not quite sure what you mean by rendering. To my knowledge tfjs doesnt do any rendering..

Yes, WebGL is used for computations

Upvotes: 4

Related Questions