optical-regeneration/notes/pytorch-install.md

pytorch install

create venv

python -m venv ./.venv
source ./.venv/bin/activate

install pytorch

https://pytorch.org/get-started/locally/

pip install torch torchvision torchaudio

test pytorch install

import torch
x = torch.rand(5, 3)
print(x)

torch.cuda.is_available()