Skip to content

Commit

Permalink
More cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
cjnolet committed Oct 13, 2023
1 parent f30a57a commit 726eaf5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docs/source/raft_ann_benchmarks.md
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@ For GPU systems, where `$DATA_FOLDER` is a local folder where you want datasets
```bash
export DATA_FOLDER=path/to/store/datasets/and/results
docker run --gpus all --rm -it -u $(id -u) \
-v $DATA_FOLDER:/data/benchmarks/ \
-v $DATA_FOLDER:/data/benchmarks \
rapidsai/raft-ann-bench:23.12a-cuda11.8-py3.10 \
"--dataset deep-image-96-angular" \
"--normalize" \
Expand All @@ -140,7 +140,7 @@ Where:
```bash
export DATA_FOLDER=path/to/store/datasets/and/results # <- local folder to store datasets and results
docker run --gpus all --rm -it -u $(id -u) \
-v $DATA_FOLDER:/data/benchmarks/ \
-v $DATA_FOLDER:/data/benchmarks \
rapidsai/raft-ann-bench:23.12a-cuda11.8-py3.10 \ # <- image to use, either `raft-ann-bench` or `raft-ann-bench-datasets`, can choose RAPIDS, cuda and python versions.
"--dataset deep-image-96-angular" \ # <- dataset name
"--normalize" \ # <- whether to normalize the dataset, leave string empty ("") to not normalize.
Expand All @@ -154,7 +154,7 @@ For CPU systems the same interface applies, except for not needing the gpus argu
```bash
export DATA_FOLDER=path/to/store/datasets/and/results
docker run all --rm -it -u $(id -u) \
-v $DATA_FOLDER:/data/benchmarks/ \
-v $DATA_FOLDER:/data/benchmarks \
rapidsai/raft-ann-bench-cpu:23.12a-py3.10 \
"--dataset deep-image-96-angular" \
"--normalize" \
Expand Down

0 comments on commit 726eaf5

Please sign in to comment.