DEV Community

Super Kai (Kazuya Ito)
Super Kai (Kazuya Ito)

Posted on

Install PyTorch and JupyterLab on Ubuntu

Buy Me a Coffee

First, update and upgrade Ubuntu:

sudo apt update && sudo apt -y upgrade
Enter fullscreen mode Exit fullscreen mode

Next, check the Python version:

python3 --version
# Python 3.12.3
Enter fullscreen mode Exit fullscreen mode

Next, install python3.12-venv:
*Memos:

  • For example, if the Python version is Python 3.13.x, you need to install python3.13-venv.
  • By default, python3.xx-venv isn't installed on Ubuntu.
sudo apt install -y python3.12-venv
Enter fullscreen mode Exit fullscreen mode

This is the current directory:

pwd
# /home/kai
Enter fullscreen mode Exit fullscreen mode

Next, create a virtual environment:

python3 -m venv venv
Enter fullscreen mode Exit fullscreen mode

Next, activate the virtual environment:

. venv/bin/activate
Enter fullscreen mode Exit fullscreen mode

Next, install PyTorch with CUDA 11.8. *There are other CUDA(Compute Unified Device Architecture) versons and a ROCm(Radeon Open Compute Platform) version and CPU version available from here:

pip install torch torchvision torchaudio
\ --index-url https://download.pytorch.org/whl/cu118
Enter fullscreen mode Exit fullscreen mode

Next, install JupyterLab:

pip install jupyterlab
Enter fullscreen mode Exit fullscreen mode

Next, open JupyterLab on Firefox:

jupyter lab
Enter fullscreen mode Exit fullscreen mode

*You can open JupyterLab with specific directory on Firefox:

jupyter lab --notebook-dir /home/kai
Enter fullscreen mode Exit fullscreen mode

Image description

Image of Timescale

Timescale – the developer's data platform for modern apps, built on PostgreSQL

Timescale Cloud is PostgreSQL optimized for speed, scale, and performance. Over 3 million IoT, AI, crypto, and dev tool apps are powered by Timescale. Try it free today! No credit card required.

Try free

Top comments (0)

👋 Kindness is contagious

Explore a sea of insights with this enlightening post, highly esteemed within the nurturing DEV Community. Coders of all stripes are invited to participate and contribute to our shared knowledge.

Expressing gratitude with a simple "thank you" can make a big impact. Leave your thanks in the comments!

On DEV, exchanging ideas smooths our way and strengthens our community bonds. Found this useful? A quick note of thanks to the author can mean a lot.

Okay