From 5486e837931c42e785b7f2f8578ec13fef15ca39 Mon Sep 17 00:00:00 2001 From: Daniel Wysocki Date: Tue, 5 Jan 2021 18:40:37 -0600 Subject: [PATCH] Editing Cupy installation instructions Most users will not want to compile Cupy from source. I have expanded the Cupy installation guide to explain how to get the version appropriate for your system. --- INSTALL.md | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/INSTALL.md b/INSTALL.md index e27170c5c..55bcc211c 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -61,10 +61,15 @@ The code also requires a working version of glue, supporting `glue.ligolw.ligolw The code uses cupy to access GPUs. If you don't have one, the code will still work. If you do need one, make sure to install cupy **on a machine that supports GPUs ** +You probably do not want to pip install `cupy` directly, but rather a pre-compiled version for your system's CUDA version. To find your system's version (assuming it's installed) run `nvidia-smi`, and look for "CUDA Version" in the top-right. For version X.Y, the appropriate cupy package will be called `cupy-cudaXY` (e.g., if you have CUDA 11.0, install `cupy-cuda110`) + ```bash -pip install --user cupy +pip install --user cupy-cudaXY # replace XY with the appropriate numbers ``` +See https://cupy.dev/ for more details. + + ## Other environment variables If you run on an LDG cluster, you need accounting flags