PyTorch with Arch Linux on AMD AI Max+ 395
Setup
install PyTorch: sudo pacman -S python-pytorch-opt-rocm
setup env: uv venv --system-site-packages --python /usr/bin/python
add marimo: uv pip install marimo[lsp] pip ty
run notebook: uv run marimo edit rocm.py
Product link.
blog.chessgrip.com1 min read