DEV Community

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

Posted on

Install PyTorch and JupyterLab on Ubuntu

#python #pytorch #jupyterlab #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

Top comments (0)

Read next

indalyadav56 profile image

Python Best Practices: Writing Clean and Maintainable Code

Indal Kumar -

velan profile image

EchoAPI vs Bruno: A Comprehensive Comparison from Design to Testing 💡

Velan<> -

ajayececit profile image

Code Your Diagrams: Automate Architecture with Python's Diagrams Library

Ajay Balakumaran -

saint_vandora profile image

Simplify Authentication with FastAPI!

FOLASAYO SAMUEL OLAYEMI -