GPU support in flight

Hello,

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

does anyone know if there is something similar for openflight?

Thankyou in advance

Kaya

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

Cheers,

Stu