From c2b997f31c3d8188697ff148ad32d483b4bca8e9 Mon Sep 17 00:00:00 2001
From: YanxuanLiu <104543031+YanxuanLiu@users.noreply.github.com>
Date: Fri, 8 Dec 2023 15:11:36 +0800
Subject: [PATCH] Remove CUDA version definition in benchmark script [skip ci]
 (#529)

* changed default cuda version from 11.8 to 11.8.0

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* for test

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* upgrade cuda versin

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* change cuda version to 12.2

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* change cuda version to 12.2.2

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* add if

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* remove cuda version

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

* add if

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>

---------

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>
---
 python/benchmark/dataproc/start_cluster.sh | 2 --
 1 file changed, 2 deletions(-)

diff --git a/python/benchmark/dataproc/start_cluster.sh b/python/benchmark/dataproc/start_cluster.sh
index 601566bd..76fa11a1 100755
--- a/python/benchmark/dataproc/start_cluster.sh
+++ b/python/benchmark/dataproc/start_cluster.sh
@@ -13,7 +13,6 @@ if [[ -z ${GCS_BUCKET} ]]; then
 fi
 
 BENCHMARK_HOME=${BENCHMARK_HOME:-${GCS_BUCKET}/benchmark}
-CUDA_VERSION=${CUDA_VERSION:-11.8}
 
 gpu_args=$(cat <<EOF
 --master-accelerator type=nvidia-tesla-t4,count=1
@@ -21,7 +20,6 @@ gpu_args=$(cat <<EOF
 --initialization-actions gs://${BENCHMARK_HOME}/spark-rapids.sh,gs://${BENCHMARK_HOME}/init_benchmark.sh
 --metadata gpu-driver-provider="NVIDIA"
 --metadata rapids-runtime=SPARK
---metadata cuda-version=${CUDA_VERSION}
 --metadata benchmark-home=${BENCHMARK_HOME}
 EOF
 )