diff --git a/context/raft-ann-bench/run_benchmark.sh b/context/raft-ann-bench/run_benchmark.sh index 66f88ece..7e56704c 100644 --- a/context/raft-ann-bench/run_benchmark.sh +++ b/context/raft-ann-bench/run_benchmark.sh @@ -14,6 +14,8 @@ function hasArg { (( ${NUMARGS} != 0 )) && (echo " ${ARGS} " | grep -q " $1 ") } +. /opt/conda/etc/profile.d/conda.sh; conda activate rapids + export CONDA_PREFIX=/opt/conda/envs/rapids export DATASET_ARG=$1 export GET_DATASET_ARGS=$2 diff --git a/context/raft-ann-bench/run_benchmarks_preloaded_datasets.sh b/context/raft-ann-bench/run_benchmarks_preloaded_datasets.sh index 25a2e8e2..c47b3e3e 100644 --- a/context/raft-ann-bench/run_benchmarks_preloaded_datasets.sh +++ b/context/raft-ann-bench/run_benchmarks_preloaded_datasets.sh @@ -14,6 +14,8 @@ function hasArg { (( ${NUMARGS} != 0 )) && (echo " ${ARGS} " | grep -q " $1 ") } +. /opt/conda/etc/profile.d/conda.sh; conda activate rapids + export CONDA_PREFIX=/opt/conda/envs/rapids export DATASET_ARG=$1 export GET_DATASET_ARGS=$2