直接安装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
版权声明:本文为zhangyichuan_dlut原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。