benchmark
c678454d - Setup CUDA environment when running bisection and abtest (#1141)

Commit
3 years ago
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
Author
Parents
Loading