Install magma on CUDA 11.1 (#48164)
Summary:
cc: xwang233 janeyx99
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48164
Reviewed By: malfet, zhangguanheng66
Differential Revision: D25058068
Pulled By: janeyx99
fbshipit-source-id: ab136ba60e5dda6a2eb7ac76548875e1df75a242
