【发布时间】:2020-01-03 12:40:57
【问题描述】:
条件
- CuPy 版本 7.0.0
- 操作系统/平台 Ubuntu 18.04
- CUDA 版本 10.1
重现代码
import cupy as np
import time
size = 60000000
tag = np.zeros(size)
#np.random.shuffle(tag)
value = np.random.random(size)
starttime = time.perf_counter()
for i in range(100):
tag +=(value> 0.3)*100
print (time.perf_counter() - starttime)
starttime = time.perf_counter()
cpu_value = np.asnumpy(value)
print (time.perf_counter() - starttime)
- 将 cupy 转换为 numpy 非常慢
结果是
0.02095769099832978 6.170492547998947
【问题讨论】: