运行如下代码
net = nn.Sequential()
net.add(nn.Dense(1))
net.initialize(ctx=mx.gpu())
net(y)
我的cunda是9.0
报错:
MXNetError: [18:02:15] c:\ci\libmxnet_1533399150922\work\src\operator\linalg_impl.h:140: Check failed: e== CUBLAS_STATUS_SUCCESS (13 vs. 0) cuBLAS: CUBLAS_STATUS_EXECUTION_FAILED
解决办法:到nvidia官网下载cunda补丁
我下载的是Patch 2 (Released Mar 5, 2018)
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork
本文记录了一位开发者在使用MXNet框架进行GPU运算时遇到的CUBLAS_STATUS_EXECUTION_FAILED错误,并详细描述了如何通过下载并安装NVIDIA官方提供的CUDA补丁来解决这一问题。
774

被折叠的 条评论
为什么被折叠?



