Difference: PreparingThePC (4 vs. 5)

Revision 52023-10-17 - UliRaich

Line: 1 to 1

META TOPICPARENT	name="TinyML"

Preparing the PC for Tensorflow

Setting up a virtual Python environment

Line: 41 to 41

TensorFlow relies on the cuda toolkit

and on the NVidia display driver. Even after installing the newest version of cuda, TensorFlow did not see the GPU. An intensive search with Google finally showed that (at the moment of writing this page) TensorFlow version 2 only works with cuda 11.8 and cuDNN 8.6.

Changed:

<
<

https://medium.com/@juancrrn/installing-cuda-and-cudnn-in-ubuntu-20-04-for-deep-learning-dad8841714d6

describes how to install cuda-11.8 and cudnn-8.6 and how to test the installation.

>
>

https://medium.com/@juancrrn/installing-cuda-and-cudnn-in-ubuntu-20-04-for-deep-learning-dad8841714d6

and https://gist.github.com/MihailCosmin/affa6b1b71b43787e9228c25fe15aeba

decribe how to install cuda-11.8 and cudnn-8.6 and how to test the installation.

The script testGPU.py below shows the number of GPUs that TensorFlow has fouind: #!/home/uli/.virtualenvs/AI/bin/python
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

View topic | History: r5 < r4 < r3 < r2 | More topic actions...