Skip to content

Instantly share code, notes, and snippets.

@nwstephens
Created September 25, 2023 21:17
Show Gist options
  • Save nwstephens/1fa31249a2958f984e3550e1d7a8665a to your computer and use it in GitHub Desktop.
Save nwstephens/1fa31249a2958f984e3550e1d7a8665a to your computer and use it in GitHub Desktop.
RAFT Benchmarks
docker pull rapidsai/staging:docker-raft-ann-bench-589-23.10a-cuda11.8-py3.9-amd64
export DATA_FOLDER=/home/rapids/benchmarks
docker run --gpus all --rm -it \
-v $DATA_FOLDER \
rapidsai/staging:docker-raft-ann-bench-589-23.10a-cuda11.8-py3.9-amd64 \
"--dataset sift-128-euclidean" \
"--normalize" \
"--algorithms raft_cagra" \
""
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment