Difference: PreparingThePC (1 vs. 2)

Revision 22022-01-31 - UliRaich

Line: 1 to 1
 
META TOPICPARENT name="TinyML"

Preparing the PC for Tensorflow

Setting up a virtual Python environment

Line: 12 to 12
 switch to a virtual environment with the "workon" command:

e.g. workon AI

Added:
>
>
The virtual environment used to build ML models should include and maybe a few others.
 

Setting up the GPU

When trying my first simple TensorFlow program I saw this warning:

Revision 12022-01-31 - UliRaich

Line: 1 to 1
Added:
>
>
META TOPICPARENT name="TinyML"

Preparing the PC for Tensorflow

Setting up a virtual Python environment

The procedure is described in https://www.freecodecamp.org/news/virtualenv-with-virtualenvwrapper-on-ubuntu-18-04.

Once his is done you can create a new virtual environment with:

mkvirtualenv name_of_env (e.g. mkvirtualenv AI)

switch to a virtual environment with the "workon" command:

e.g. workon AI

Setting up the GPU

When trying my first simple TensorFlow program I saw this warning:

libcudart.png

My PC, being a "gaming machine", uses an nVIDIA GeForce GTX 950M graphics card whose GPU (Graphics Processing Unit) can be used by TensorFlow to speed up number crunching.

nvidia.png

TensorFlow relies on the cuda toolkit and on the NVidia display driver. I had difficulties to install these with the .deb files but succeeded to do so using the .run file.

cuda_download.png

To run the installer script I had to boot into single user mode, avoiding to start the desktop system. This can be done typing "e" at the boot screen and adding "single" to the boot command.

In the "Additional Drivers" tab of the Software & Updates program you see that we are using the manually installed NVidia driver:

driver_used.png

When trying a very simple TensorFlow program: #!/usr/bin/python3
import tensorflow as tf
import numpy as np
from tensorflow import keras
model = tf.keras.Sequential([keras.layers.Dense(units=1, input_shape=[1])])

I get:

gpu_good.png

-- Uli Raich - 2022-01-31

Comments

<--/commentPlugin-->

META FILEATTACHMENT attachment="cuda_download.png" attr="" comment="" date="1643626582" name="cuda_download.png" path="cuda_download.png" size="52637" user="UliRaich" version="1"
META FILEATTACHMENT attachment="nvidia.png" attr="" comment="" date="1643626582" name="nvidia.png" path="nvidia.png" size="13000" user="UliRaich" version="1"
META FILEATTACHMENT attachment="driver_used.png" attr="" comment="" date="1643626582" name="driver_used.png" path="driver_used.png" size="76697" user="UliRaich" version="1"
META FILEATTACHMENT attachment="libcudart.png" attr="" comment="" date="1643626913" name="libcudart.png" path="libcudart.png" size="37871" user="UliRaich" version="1"
META FILEATTACHMENT attachment="gpu_good.png" attr="" comment="" date="1643626582" name="gpu_good.png" path="gpu_good.png" size="46971" user="UliRaich" version="1"
META FILEATTACHMENT attachment="graphicsCard.png" attr="" comment="" date="1643626913" name="graphicsCard.png" path="graphicsCard.png" size="39171" user="UliRaich" version="1"
 
This site is powered by the TWiki collaboration platform Powered by PerlCopyright © 2008-2024 by the contributing authors. All material on this collaboration platform is the property of the contributing authors.
Ideas, requests, problems regarding TWiki? Send feedback