安装gpu版本 mxnet
WIN10+ GTX1050CUDA8.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...
·
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 完美解决上述问题。。。
权威｜前沿｜技术｜干货｜国内首个API全生命周期开发者社区
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)