GPU support in flight


GPU support doesn’t work automatically for TensorFlow on my OpenFlight in Azure. I’ve been using this guide to install on Ubuntu:

does anyone know if there is something similar for openflight?

Thankyou in advance


Hi Kaya,

From the looks of things, the GPU version is just a slightly different package name. Sadly (presumably down to the containerised nature of Singularity) I cannot seem to find a decent image for running Tensorflow GPU through it, however, Conda seems to be able to install it.

To use GPU Tensorflow with Conda (replacing 1.15 with your desired version of tensorflow):

pip install tensorflow-gpu==1.15