PyCuda配置（二）下载PyCuda

一 命令行安装

比较简单的方法是通过apt-get下载，调用如下命令：

sudo apt-get install python-pycuda

这样可以安装python2对应版本的pycuda。

安装后用以下代码验证一下即可。

#!/usr/bin/env python

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy as np

from datetime import datetime

a = np.random.randn(4,4)
a = a.astype(np.float32)

mod = SourceModule("""
  __global__ void doublify(float *a)
  {
    int idx = threadIdx.x + threadIdx.y*4;
    a[idx] *= 2;
  }
  """)

startTime = datetime.now()
# CUDA method
a_gpu = cuda.mem_alloc(a.nbytes)
cuda.memcpy_htod(a_gpu, a)
a_doubled = np.empty_like(a)
cuda.memcpy_dtoh(a_doubled, a_gpu)
print "Cuda time is: ", datetime.now()-startTime
print a_doubled

startTime = datetime.now()
# np method
a_doubled_2 = a * 2
print "np method time is: ", datetime.now()-startTime
print a_doubled_2

 二 下载源代码编译安装

1 下载pycuda

2 命令行进行如下指令

tar xzvf pycuda-VERSION.tar.gz
cd pycuda-VERSION
python configure.py --cuda-root=/where/ever/you/installed/cuda
或
python3 configure.py --cuda-root=/where/ever/you/installed/cuda

注意如果是安装python3的话，一般上述命令中的python要改为python3

 

问题

1 是否需要第一步中的安装CUDA？　　　　->　　用一台未配置的电脑试试

2 例子中的加速函数参数等应当如何配置？　　->　　继续学习pycuda tutorial

 

Credits:

https://blog.csdn.net/JohnJim0/article/details/100585885

https://blog.csdn.net/wj164/article/details/45648517

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