window10下使用MXNet(GPU)运行神经网络报错:Check failed: e == CUBLAS_STATUS_SUCCESS (13 vs. 0) cuBLAS: CUBLAS_STA
运行如下代码
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: CUBLA
原创
2020-07-17 19:38:55 ·
1062 阅读 ·
0 评论