Fix regex in collect_env.py for CUDA 11 (#51852)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/51840
Manually tested with both CUDA 10.2.89 & 11.2.67.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51852
Reviewed By: izdeby
Differential Revision: D26326105
Pulled By: mrshenli
fbshipit-source-id: 46fbe5f20c02bca982ce2ec6e62f7cc3a14fcc97
