未找到cupy卷积

Question

我正在尝试获取cupy来使用https://docs.cupy.dev/en/latest/reference/generated/cupy.convolve.html文档中的convolve。

>>> import cupy as cp
>>> b = cp.array([0,0,0,1,0,0,0,1])
>>> cp.convolve(b, cp.ones(3), 'same') / 3
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'cupy' has no attribute 'convolve'

因此，对numpy进行权力下放似乎有效，但对cupy则不起作用。

>>> import numpy as np
>>> a = np.array([0,0,0,1,0,0,0,1])
>>> np.convolve(a, np.ones(3), 'same') / 3
array([0.        , 0.        , 0.33333333, 0.33333333, 0.33333333,
       0.        , 0.33333333, 0.33333333])

虽然基本功能如添加与cupy工作。

>>> cp.add(b, b)
array([0, 0, 0, 2, 0, 0, 0, 2], dtype=int32)

设置：

• Windows 10
• Python3.8.5(标签/v3.8.5:580 fbb0,2020年7月20日，15:57:54) MSC v.1924 (AMD64) on win32
• cuda 11.0.3 451.82

cuDNN 11.0 x64 1.2.0F 215

和平执行方案：

• numpy-1.19.1
• cupy-cuda110-7.8.0
• optuna-2.0.0

我对Python和cuda进行了全新的安装，.，但这并没有解决问题。我现在不知道还有什么可能会丢失。据我所知，numpy只是在gpu上处理，所以numpy的每个功能都应该在cupy中。

Answer 1

因为cupy 7.8.0似乎不支持convolve，只支持8.0.0版本，所以我现在安装了当前的候选版本cupy_cupy 110-8.0.0rc1。

pip install --pre -U cupy-cuda110

现在convolve运作得很好