Installing PyTorch on Ubuntu

PyTorch is an open-source machine learning library widely used for deep learning applications. This guide will walk you through the process of installing PyTorch on Ubuntu and provide an overview of how to use it for your machine learning projects.

we will need Python installed (3.6 or later is recommended), if you did not install it you can follow do that like this:

The installation of PyTorch may vary depending on whether you want to use it with a CPU or GPU. The official PyTorch website provides a straightforward installation command generator based on your preferences.

1- Visit the PyTorch website: PyTorch Get Started.

2- Select your preferences and choose your desired configuration:

3- Copy the installation command provided by the website. It will look something like this:

For CPU:

For CUDA (GPU):

To verify that PyTorch has been installed correctly, you can run the following command in a Python shell:

Then, within the Python shell, type:

If you see the version number of PyTorch printed without any errors, your installation was successful!

Now that you have PyTorch installed, let’s create a simple script to test its functionality.

1- Create a new Python file (e.g., test_pytorch.py):

2- Add the following code to the file:

3- Run your script:

You should see output similar to the following, confirming that PyTorch is working:

You have successfully installed PyTorch on your Ubuntu system and ran a simple script to verify its functionality. PyTorch is a powerful library that enables you to build and train deep learning models efficiently. You can now explore more advanced features and start developing your machine learning projects using PyTorch!