cupy 相关问题


Cupy `RawKernel` CUDA_ERROR_NOT_FOUND:未找到命名符号 [cupy]

我正在尝试使用cupy的RawKernel在python中编写自定义cuda内核,但是我不断收到以下错误 回溯(最近一次调用最后一次): 文件“/nfs/users/xxxxxxxxx/git/raw_k...


CUDA_ERROR_INVALID_SOURCE:设备内核映像无效

我正在尝试在我的 docker 容器中使用 cupy。 我使用的容器其中一个用于 CUDA 和 cuDNN,另一个用于 cupy。 我尝试了这段代码。 导入 cupy 作为 cp cupy_array = cp.array([1, 2, 3])


如何在Cupy内核中使用WMMA函数?

如何在cupy.RawKernel或cupy.RawModule中使用wmma::load_matrix_sync等WMMA函数? 有人可以提供一个最小的例子吗?


© www.soinside.com 2019 - 2024. All rights reserved.