How to Install TensorFlow on Ubuntu 24.04

Once installed, you can verify Python and pip installation before installing packages like TensorFlow.

python3 --version
pip3 --version

Step 2: Create a Virtual Environment in Ubuntu

Creating a virtual environment is optional but recommended, as it helps to keep your TensorFlow installation isolated from other Python projects.

Create a new directory for your TensorFlow project and navigate into it:

mkdir tensorflow_project
cd tensorflow_project

Create a virtual environment and activate it:

python3 -m venv tensorflow_env
source tensorflow_env/bin/activate

Your terminal prompt should now show the name of the virtual environment, indicating that it is active.

Step 3: Install TensorFlow in Ubuntu

Now that your environment is set up, you can install the latest stable version of TensorFlow along with its dependencies using pip, the Python package manager.

pip install --upgrade pip
pip install tensorflow

Step 4: Verify TensorFlow in Ubuntu

After the installation is complete, it’s a good idea to verify that TensorFlow is installed correctly by going to Python shell.

python3

Import TensorFlow and check its version:

import tensorflow as tf
print(tf.__version__)

If TensorFlow is installed correctly, this will print the version number without any errors.

2.18.0

You can also run a simple test to ensure TensorFlow is working.

hello = tf.constant('Hello, TensorFlow!')
print(hello)

This should output something like.

tf.Tensor(b'Hello, TensorFlow!', shape=(), dtype=string)

Step 5: Install TensorFlow with GPU Support (Optional)

If you have a compatible NVIDIA GPU and want to use it for faster computation, you can install TensorFlow with GPU support.

lspci | grep -i nvidia
sudo apt install nvidia-driver-535 -y
sudo reboot

Next, install TensorFlow with GPU support.

pip install tensorflow-gpu

Verify the installation by checking if TensorFlow detects your GPU:

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

Conclusion

You have successfully installed TensorFlow on Ubuntu 24.04. Whether you are working on a simple machine learning project or a complex deep learning model, TensorFlow provides the tools you need to get started.

Remember to keep your environment organized by using virtual environments, and consider GPU support if you need faster computation.