环境:
- Ubuntu 14.04
- 依赖项 Numpy
1. PyCUDA 安装
1.1 直接安装
sudo apt install python-pycuda
1.2 源码安装：
- Step 1 下载并解压 PyCUDA
$ wget http://pypi.python.org/pypi/pycuda
$ tar xfz pycuda-VERSION.tar.gz
- Step 2 编译PyCUDA
$ cd pycuda-VERSION # if you're not there already
$ python configure.py --cuda-root=/where/ever/you/installed/cuda
$ su -c "make install"
2. PyCUDA 测试
- e.g. 1
$ cd pycuda-VERSION/test
$ python test_driver.py
输出 OK 即成功.
- e.g. 2
$ cd pycuda-VERSION/examples
$ python test_demo.py
- e.g. 3
import pycuda.autoinit
import pycuda.driver
free_bytes, total_bytes = pycuda.driver.mem_get_info() # 查询当前显卡的总显存，可用显存