Hands-On GPU Computing with Python

pip-based installation of CuPy

After going through the Conda-based installation of CuPy, let's explore another method that also can be used. You can use pip to install CuPy with the local environment. The following code demonstrates how you can do this:

$ pip3 install --upgrade pip$ pip3 install --user cupy-cuda90

Note that pip install --user cupy-cuda90 performs the same operation as pip3 install --user cupy-cuda90. The command for CUDA 10.0 is pip3 install cupy-cuda100.

The alternative command for installing CuPy for Python 2.x would be the following:

$ pip2 install --upgrade pip$ pip2 install --user cupy-cuda90

Note that the preceding command is specific for CUDA 9.0. CuPy also supports versions 8.0, 9.1, 9.2, and 10.0.

Using just cupy ...