Install time (#2493)

Summary: Remove nvidia-cuda-nvcc-cu12 as not required. Install time. Pull Request resolved: #2493 Reviewed By: xuzhao9 Differential Revision: D63987509 Pulled By: atalman fbshipit-source-id: 07298ddb569da7f7c3fe22d73da72a4ceab256f5
pytorch · Oct 7, 2024 · eae9e50 · eae9e50
1 parent bde2401
commit eae9e50
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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"