diff --git a/.github/workflows/torchbench.yml b/.github/workflows/torchbench.yml index dafd8d2c6782..3ee0ab6fc007 100644 --- a/.github/workflows/torchbench.yml +++ b/.github/workflows/torchbench.yml @@ -43,17 +43,17 @@ jobs: df -h du --max-depth=1 -h - # - name: Tune Nvidia GPU - # run: | - # sudo nvidia-smi -pm 1 - # sudo nvidia-smi --lock-gpu-clocks=1280,1280 - # nvidia-smi + - name: Tune Nvidia GPU + run: | + sudo nvidia-smi -pm 1 + sudo nvidia-smi --lock-gpu-clocks=1280,1280 + nvidia-smi - name: Run TorchBench run: | cd torchbench python3 test.py -v TestBenchmark.test_detectron2_fasterrcnn_r_101_c4_train_cuda - # - name: Reset GPU clocks - # run: | - # sudo nvidia-smi -i 0 -rgc + - name: Reset GPU clocks + run: | + sudo nvidia-smi -i 0 -rgc