执行:pip install pycuda
出现:Building wheel for pycuda (PEP 517) ... error
使用:pip install pycuda --no-binary :all:
不解决问题。
从源码编译,解决问题。
1、下载 pycuda-2021.1.tar.gz
2、tar xfz pycuda-2021.1.tar.gz
3、cd pycuda-2021.1
4、python3 configure.py --cuda-root=/usr/local/cuda
5、make install
如果有找不到 pyconfig.h 的错误,就安装 python3-dev。
如果有找不到 nvcc 的错误,就 export CUDA_ROOT=/usr/local/cuda/bin。
参考:
https://wiki.tiker.net/PyCuda/Installation/Linux/
https://stackoverflow.com/questions/8368815/pycuda-cuda-root-not-set-and-nvcc-not-in-path