-r requirements.txt

# Get hardware-appropriate torch/torchvision 
--extra-index-url https://download.pytorch.org/whl/rocm5.1.1 --trusted-host https://download.pytorch.org
torch
torchvision
-e .