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

Retry later

Top comments (0)

Sentry image

See why 4M developers consider Sentry, “not bad.”

Fixing code doesn’t have to be the worst part of your day. Learn how Sentry can help.

Learn more

Retry later