Setup CUDA environment when running bisection and abtest (#1141)
Summary:
CUDA 11.3 is being deprecated, and we are using environment variables to control the CUDA version instead of changing the link target of `/usr/local/cuda`.
This has many benefits:
- switching cuda version no longer needs sudo permission
- support switching cuda versions at both runtime and compile time
This PR also creates a `cuda_utils.py` to place all CUDA version related code.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1141
Reviewed By: erichan1, davidberard98
Differential Revision: D39139827
Pulled By: xuzhao9
fbshipit-source-id: 6c305ab749c8cfc0d5bc8f15a86dc4a0a6d04a96