直接安装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
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)