Jupyter Notebook is a tool to run and write Python code easily, showing results right away, and allowing you to combine code, charts, notes, and files in one place ...
Tested with Python 3.10.13, PyTorch 2.5.1, CUDA 12.6. Dependencies are captured in two files at the repo root: environment.yml (conda — canonical) and requirements.txt (pip-only pins).