ModuleNotFoundError: No module named ‘pycuda‘

环境:ubuntu 16.04

可通过以下指令直接安装:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pycuda --no-binary :all:

如果通过上述命令安装过程出现如下问题:

ERROR: Could not build wheels for pycuda, which is required to install pyproject.toml-based projects

是因为通过 pip 默认安装的都是最新版本,而最新版的 pycuda 可能和 cuda 版本不一致,因为 pycuda 更新速度较慢,只适配了部分版本的 cuda 。

可通过下载对应自己系统 cuda 版本的 pycuda 的.whl 文件,手动安装,下载地址:

https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycudaicon-default.png?t=M1H3https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

你可能感兴趣的:(PyTorch,python,深度学习,pycuda,安装pycuda)