Lambda stack versions of cuda and Tensorflow

Tensorflow 2.5 requires cuda 11.2. My laptop has cuda 10.0 installed. If I upgrade using Lambda Stack, which versions of cuda and Tensorflow will be installed?

With Ubuntu 20.04 LTS:

  • NVIDIA Driver 11.2 with support of up to CUDA 11.2
  • CUDA Toolkit 11.1

And it is Tensorflow 2.5.0:
$ TF_CPP_MIN_LOG_LEVEL=3 python -c ‘import tensorflow as tf ; print(“Tensorflow Version:”,tf.version)’
Tensorflow Version: 2.5.0

I hope that helps.

1 Like