运行py文件时报错:ModuleNotFoundError: No module named 'cupy’以及pynvrtc库的安装

安装cupy库:

可以通过pip或conda命令来安装cupy,这里我使用的pip命令:

1
pip install cupy-cuda116

实际上我的cuda版本是11.8,但并没有对应的118版本的cupy,所以我选择了cupy-cuda116,其他版本的cuda可以根据自己的cuda版本选择。

安装pynvrtc库:

运行 nvcc --version 或 nvidia-smi 查看 CUDA 版本。
如果 CUDA ≥11.0,可能需要手动安装旧版 pynvrtc:

1
pip install pynvrtc==9.2