Logo

Indie Machine Learning and Video Game Dev

Neural Net Downloads
How to
Useful links
Hardware Reviews
Get Merch
About

Check if using tensorflow gpu version

by Mathieu Poliquin

tf.test.is_built_with_cuda()

Note: not sure how this funciton behaves with rocm

tags: tensorflow - gpu - version - check