Installing PyTorch on Ubuntu 24.04: A Step-by-Step Guide

Install PyTorch using pip

You can install PyTorch using pip on Ubuntu 24.04 by first updating your system packages and getting Python. Run the command sudo apt update && sudo apt install python3 python3-pip in your terminal. This gets your system ready to download PyTorch and the tools it needs to run properly.

Isolated virtual environments keep project dependencies separate and prevent version conflicts across different Python projects.

This isolated environment keeps PyTorch and its dependencies separate, preventing conflicts with other Python projects on the system.

Create a Python virtual environment for PyTorch

You need to create a Python virtual environment for PyTorch to keep your project files separate from other Python programs on your computer. Run sudo apt install virtualenv in your terminal to get the tools you need. This stops different package versions from breaking each other when you install PyTorch.

Install the 💿 required package management tools by running `sudo apt install virtualenv`.

sudo apt install virtualenv

Set up a dedicated workspace named myenv by running the following command.

virtualenv myenv

Activate the workspace using this command.

source myenv/bin/activate

Running this command installs PyTorch inside the active environment.

pip3 install torch torchvision torchaudio

Launch the Python shell and import PyTorch after the installation finishes.

python
import torch

Check the installed version by printing it.

print(torch.__version__)

A successful installation outputs the version number.

Deactivate the virtual environment using the command below when work concludes.

deactivate

Install PyTorch using Anaconda

You can install PyTorch using Anaconda on Ubuntu 24.04 as an alternative to pip for managing your Python packages. Download the Miniconda installer by running curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh in your terminal. This approach keeps your PyTorch setup self-contained and easy to manage.

Visit the Anaconda website to fetch the Linux-x86_64 installer matching system requirements.

Execute this command to fetch the installer script.

curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

Run the downloaded script to complete the Anaconda setup.

bash Miniconda3-latest-Linux-x86_64.sh

Accept the license terms and default installation location by typing yes when prompted by the script.

Configure the shell environment for Anaconda execution.

source ~/.bashrc
conda config --set auto_activate_base false

Install PyTorch by executing the following command.

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Add GPU support (requires a compatible NVIDIA GPU and CUDA toolkit):

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

Change pytorch-cuda=11.7 to match the specific CUDA version installed on the machine.

Open the Python shell and import PyTorch.

python
import torch

Verify the installation by printing the version number.

print(torch.__version__)

Successful execution displays the installed version.

Configuration is now complete.

Conclusion:

PyTorch installation on Ubuntu 24.04 is now complete using either pip or Anaconda. Key takeaways include:

• Flexible Installation Methods: Choose pip for simplicity or Anaconda for advanced environment management.
• Virtual Environments: Using a virtual environment helps maintain project dependencies and prevents conflicts.
• GPU Acceleration: Integrate PyTorch with CUDA for enhanced training performance on NVIDIA GPUs.
• Version Verification: Always check the installed PyTorch version to ensure your setup is correct.
• Community and Support: PyTorch has a large community and extensive documentation, making it a great choice for deep learning projects.

Following these steps establishes a working PyTorch environment ready for artificial intelligence and deep learning projects.