Failed to import CuPy ImportError: libcudart.so.10.2: cannot open shared object file: No such file

import cupy报错解决方法

Failed to import CuPy ImportError: libcudart.so.10.2: cannot open shared object file: No such file

Failed to import CuPy ImportError: libcudart.so.10.2: cannot open shared object file: No such file_第1张图片

有的算法需要安装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即可

你可能感兴趣的:(Debug,python,深度学习,pytorch)