安装pycuda的正确办法

  • 参考

https://wiki.tiker.net/PyCuda/Installation/Linux
 

使用pip install pycuda,出现找不到cuda.h。没解决。下载、解压

  • 编译安装

下载源码在python 3.5 下make,一直提示说找不到numpy。后来也不知道怎么解决了。

# 注意是python 是2.7/3.5
./configure.py --python-exe=/usr/bin/python3 --cuda-root=/usr/local/cuda --cudadrv-lib-dir=/usr/lib/x86_64-linux-gnu --boost-inc-dir=/usr/include --boost-lib-dir=/usr/lib --boost-python-libname=boost_python-py35 --boost-thread-libname=boost_thread --no-use-shipped-boost

python3 configure.py --cuda-root=/usr/local/cuda

sudo python3 setup.py install

make -j8

sudo pip3 install .

猜你喜欢

转载自blog.csdn.net/quantum7/article/details/83548723
今日推荐