WIN10+ GTX1050 CUDA8.0==1.2.1版本 会出现一些问题。

In [1] x = nd.array([1, 2, 3],ctx=mx.gpu())
MXNetError: [12:21:47] c:\jenkins\workspace\mxnet-tag\mxnet\src\storage./pooled_storage_manager.h:108: cudaMalloc failed: device kernel image is invalid

在虚拟环境下,使用豆瓣源下载
pip install --pre mxnet-cu80==1.2.0 -i https://pypi.douban.com/simple

降级MXNET后,上述问题得以解决。。但是有些最近的API不能使用了,
在这里插入图片描述

卸载mxnet-cu80, 重新安装mxnet-cu80=1.3.0 完美解决上述问题。。。

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐