Skip to content

Commit

Permalink
remove cuda version
Browse files Browse the repository at this point in the history
Signed-off-by: YanxuanLiu <[email protected]>
  • Loading branch information
YanxuanLiu committed Dec 8, 2023
1 parent 902ff2c commit 8f638e6
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 3 deletions.
1 change: 0 additions & 1 deletion .github/workflows/gcs-benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@ on:

jobs:
Benchmark:
if: github.repository == 'NVIDIA/spark-rapids-ml'
runs-on: ubuntu-latest
env:
PROJECT: rapids-spark
Expand Down
2 changes: 0 additions & 2 deletions python/benchmark/dataproc/start_cluster.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,15 +13,13 @@ if [[ -z ${GCS_BUCKET} ]]; then
fi

BENCHMARK_HOME=${BENCHMARK_HOME:-${GCS_BUCKET}/benchmark}
CUDA_VERSION=${CUDA_VERSION:-12.2.2}

gpu_args=$(cat <<EOF
--master-accelerator type=nvidia-tesla-t4,count=1
--worker-accelerator type=nvidia-tesla-t4,count=1
--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
)
Expand Down

0 comments on commit 8f638e6

Please sign in to comment.