安装gpu版本 mxnet

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 完美解决上述问题。。。

版权声明：本文为weixin_35364049原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。