[doc] pytorch native amp leak fix landed in 1.7.1
update README with good news that the leak fix has been applied to pytorch-1.7.1 and not just 1.8.

Reference: https://github.com/pytorch/pytorch/issues/48049#issuecomment-742790722

@LysandreJik 