I want to run
on
A100 80GB