未能导入cupy

Question

在通过"pip install cupy -cupy 110“安装cupy之后，我在python3中尝试了这样的方法：

导入cupy作为cp

然而，它失败了：

$ python3 Python3.8.10(默认，11月26日2021，20:14:08) GCC 9.3.0在linux类型“帮助”、“版权”、“信用”或“许可”上获得更多信息。

将cupy导入为cp回溯(最近一次调用)："/usr/local/lib/python3.8/dist-packages/cupy/init.py"，第18行，从

导入_core # NOQA文件的第1行，在cupy._core导入核心# NOQA ImportError: libnvrtc.so.11.0:无法打开共享对象文件:没有这样的文件或目录

上述异常是导致以下异常的直接原因：

追溯(最近一次调用)：File ""，第1行，在File "/usr/local/lib/python3.8/dist-packages/cupy/init.py"，第20行中，在raise (f‘ImportError)中：

未能导入CuPy。

如果您通过轮子( CuPy -cudaXXX或cupy- ROCm )安装了cupy，请确保包与安装的CUDA或ROCm版本相匹配。

在Linux上，您可能需要根据安装CUDA/ROCm的方式设置LD_LIBRARY_PATH环境变量。在Windows上，尝试设置CUDA_PATH环境变量。

有关详细信息，请参阅安装指南：https://docs.cupy.dev/en/latest/install.html

原始错误: ImportError: libnvrtc.so.11.0:无法打开共享对象文件:没有这样的文件或目录

“

你能知道如何解决这个问题吗？非常感谢！

Answer 1

在安装cupy之前，您应该检查安装在设备上的Cuda版本：

!nvcc --version

对我来说，在我的设备上安装的Cuda版本是11.0 (见下图)：

​

然后转到官方的丘比特网站https://docs.cupy.dev/en/stable/install.html

下面的图片来自我上面提到的页面：

​

然后键入与您的Cuda版本相匹配的install命令，对于我来说，install命令是：

pip install cupy-cuda110