From eae9e50bbeff81d8eecd260f89fbbd160fabcc2f Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 7 Oct 2024 15:45:55 -0700 Subject: [PATCH] Install time (#2493) Summary: Remove nvidia-cuda-nvcc-cu12 as not required. Install time. Pull Request resolved: https://github.com/pytorch/benchmark/pull/2493 Reviewed By: xuzhao9 Differential Revision: D63987509 Pulled By: atalman fbshipit-source-id: 07298ddb569da7f7c3fe22d73da72a4ceab256f5 --- userbenchmark/release-test/run_release_test.sh | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/userbenchmark/release-test/run_release_test.sh b/userbenchmark/release-test/run_release_test.sh index 38e8141d37..68fd4d452f 100644 --- a/userbenchmark/release-test/run_release_test.sh +++ b/userbenchmark/release-test/run_release_test.sh @@ -16,12 +16,11 @@ fi . switch-cuda.sh "${CUDA_VERSION}" -if [[ ${CUDA_VERSION} == "12.1" ]]; then - pip install nvidia-cuda-nvcc-cu12 -fi nvcc --version sudo apt-get install bc +sudo apt-get install --reinstall time +which time # run mnist mkdir -p "${RESULT_DIR}/mnist" pushd "${EXAMPLES_DIR}/mnist"