CuPyインストールガイド

環境

OS: Ubuntu 16.04 LTS
CUDA: 9.0

$ cat /etc/os-release
NAME="Ubuntu"
VERSION="16.04.5 LTS (Xenial Xerus)"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 16.04.5 LTS"
VERSION_ID="16.04"
HOME_URL="http://www.ubuntu.com/"
SUPPORT_URL="http://help.ubuntu.com/"
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"
VERSION_CODENAME=xenial
UBUNTU_CODENAME=xenial

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176

インストール手順

インストール前に一応pipとsetuptoolsをアップデート

pip install -U setuptools pip

CUDAのversion確認

nvcc -V

CUDAのversion(下記は9.0の場合)のCuPyインストール

pip install cupy-cuda90

CuPyが入ってるか確認

pip freeze

cudaが入っていない or cudaをUpgradeしたい場合の参考
https://qiita.com/1O1/items/c7b0cec503d48b2efde1

参考

https://docs-cupy.chainer.org/en/stable/install.html#using-pip