import numpy as np
import cupy as cp
import time
#安装指定的cuda版本的 cp
### Numpy and CPU
s = time.time()
x_cpu = np.ones((100,100,100))
e = time.time()
print("cpu",e - s)
### CuPy and GPU
s = time.time()
x_gpu = cp.ones((100,100,100))
e = time.time()
print("gpu",e - s)
参考:
https://blog.csdn.net/qq_41185868/article/details/103479683