MXNET Windows +Anaconda下安装教程及Bug解决
前言
MXNET 作为2015年底开始流行的Deep Learning 开源框架, 由于其灵活性、对多种语言的支持、 轻量级的调度引擎已经成为很多DL 研发、学习者的开发框架。 网上也出现了很多的mxnet安装教程,但随着mxnet的更新迭代,较早的安装教程会出现一些问题。 笔者在安装过程中出现过很多头疼的Bug, 故针对mxnet 0.9.2(2017年2月更新的mxnet)在windows下的安装写一篇教程, 并汇总一些安装过程中遇到的bug,不定期更新。
1. 安装环境
1 安装Cuda
在NVidia官网下载
一定要检查你的显卡是否能用cuda, 如果不能,是用不了mxnet的 GPU版的,请安装CPU版
2 安装Cudnn
Cudnn一定要下载跟cuda对应的版本, 现在cuda 8.0对应的是cudnn 5。注意,有的较早教程提供的是7.5的Cuda,如果安错了对应的Cudnn以后会出现无法使用Cudnn的Bug
下载地址可以去 Cudnn, 要注册英伟达账号
下载后得到Cudnn文件夹(名字也叫cuda), 分别将里面的cuda/include、cuda/lib、cuda/bin三个目录中的内