PyCuda配置（二）下载PyCuda

sudo apt-get install python-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