安装pycuda的正确办法

• 参考

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

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

 

 

• 编译安装

# sudo apt-get -y --force-yes install python-pycuda
sudo rm -rf pycuda-2018.1.1
tar xzf pycuda-2018.1.1.tar.gz

cd pycuda-2018.1.1

# where is distribute_setup.py ?
su -c "python distribute_setup.py"
su -c "easy_install numpy"

python2 configure.py --cuda-root=/usr/local/cuda
su -c "make install"

cd test
python2 test_driver.py

cd ../..