PyCUDA - Ubuntu14.04安装及测试

安装

• 依赖 Numpy
• 直接安装

@H_404_10@sudo apt install python-pycuda

• 源码安装：

@H_403_18@Step 1 下载并解压 PyCUDA @H_404_10@$ wget http://pypi.python.org/pypi/pycuda $ tar xfz pycuda-VERSION.tar.gz

Step 2 编译PyCUDA

@H_404_10@$ cd pycuda-VERSION # if you're not there already $ python configure.py --cuda-root=/where/ever/you/installed/cuda $ su -c "make install"

测试

e.g. 1

@H_404_10@$ cd pycuda-VERSION/test $ python test_driver.py

输出 OK 即成功.

e.g. 2

@H_404_10@$ cd pycuda-VERSION/examples $ python test_demo.py

e.g. 3

@H_404_10@import pycuda.autoinit import pycuda.driver free_bytes,total_bytes = pycuda.driver.mem_get_info() # 查询当前显卡的总显存，可用显存

文档

官方文档