Skip to content

Editing Cupy installation instructions #47

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
dwysocki wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Editing Cupy installation instructions #47

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion INSTALL.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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`)
Copy link
Collaborator

@leofang leofang Jan 6, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is strictly speaking not accurate. CUDA Version here refers to the maximal CUDA Toolkit version supported by the driver version reported on its left, but what you really need is the runtime library version installed locally, which can be decoupled/fall behind this version (say you have CUDA 10.2 installed, but you upgrade the driver to 11.1). Perhaps it's better to use nvcc --version if there is a local installation available to check.

FYI in some environments where there is no CUDA Toolkit installed locally, but has the driver installed, you can install CuPy from Conda-Forge:

conda install -c conda-forge cupy cudatoolkit=X.Y

X.Y is again up to the driver version. The cudatoolkit contains all CUDA runtime libraries (but no nvcc or the compiler toolchain).


```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
Expand Down