Anyone know if Tensorflow Lite has GPU support for Python? I've seen guides for Android and iOS, but I haven't come across anything about Python. If tensorflow-gpu
is installed and tensorflow.lite.python.interpreter
is imported, will GPU be used automatically?
We're happy to announce that it now supports the GPU delegate. This means you can now execute your models on GPU without having to bundle GPU libraries to your app. We expect savings up to 5MB in APK size by migrating to TensorFlow Lite in Play Services, if you were using the bundled TFLite and GPU delegate on Android.
Hardware requirements Note: TensorFlow binaries use AVX instructions which may not run on older CPUs. The following GPU-enabled devices are supported: NVIDIA® GPU card with CUDA® architectures 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 and higher. See the list of CUDA®-enabled GPU cards.
TensorFlow supports running computations on a variety of types of devices, including CPU and GPU.
According to this thread, it is not.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With