benchmark
11f7141d - Upgrade to CUDA 11.7 and remove V1 workflows (#1398)

Commit
3 years ago
Upgrade to CUDA 11.7 and remove V1 workflows (#1398) Summary: Define the default Python version in `python_utils.py`. Define the default CUDA version in `cuda_utils.py`. To upgrade CUDA version: 1. Change the default CUDA version in `utils/cuda_utils.py` 2. Change the default CUDA version in the Dockerfile. Fixes https://github.com/pytorch/benchmark/issues/1390 Pull Request resolved: https://github.com/pytorch/benchmark/pull/1398 Reviewed By: atalman Differential Revision: D43086532 Pulled By: xuzhao9 fbshipit-source-id: 04b99af060be7dec951ac23558cef01e99de4472
Author
Parents
Loading