关于‘cupy’报错
运行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 |
评论