CuPy is an implementation of a NumPy-compatible multi-dimensional array on CUDA.
conda install -c rapidsai cupy
conda install -c "rapidsai/label/cuda10.0" cupy
conda install -c "rapidsai/label/cuda10.1" cupy
conda install -c "rapidsai/label/cuda9.2" cupy
conda install -c "rapidsai/label/thirdparty" cupy