Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

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?

like image 401
Sai Raghuram Kaligotla Avatar asked Oct 15 '22 13:10

Sai Raghuram Kaligotla


1 Answers

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

Yes, WebGL is used for computations

like image 95
Stewart_R Avatar answered Nov 15 '22 06:11

Stewart_R