直接安装cupy会特别的慢,甚至以为卡死了
直接安装对应的cupy-cuda包即可
- cupy-cuda101 (for CUDA 10.1)
- cupy-cuda100 (for CUDA 10.0)
- cupy-cuda92 (for CUDA 9.2)
- cupy-cuda91 (for CUDA 9.1)
- cupy-cuda90 (for CUDA 9.0)
- cupy-cuda80 (for CUDA 8.0)
比如我用的cuda10.1则直接
pip install cupy-cuda101