import cupy报错解决方法
Failed to import CuPy ImportError: libcudart.so.10.2: cannot open shared object file: No such file
有的算法需要安装cupy(一种类似numpy,使用CUDA的GPU硬件加速)
在pip安装cupy后有可能会遇到这个错误
这个错误是由于cupy版本和系统安装的CUDA版本不一致导致的
我报错的原因在于安装的cupy版本和torch对应的cuda版本对应,torch安装的是1.10.2+cu113,所以我cupy安装的是cupy-cuda113
而服务器上nvcc -V的结果是
解决方法
将原来的cupy-cuda113 pip uninstall 卸载,重新pip install cupy-cuda101即可