为mxnet配置cuda 9.2

1,安装cuda

按照《动手学深度学习》的书本和社区的建议,我下载的是9.2版本的cuda。下载链接,https://developer.nvidia.com/cuda-92-download-archive。安装的话直接跟着提示走就好,然后再控制面板看到cuda就说明安装完成。

为mxnet配置cuda 9.2

2,更新环境

为mxnet配置cuda 9.2

将原本的“mxnet”改为“mxnet-cu92”。92即为cuda版本,如果安装其他版本的cuda,那也要相应修改。更新过程中我使用的是清华的源,但也timed out很多次。无奈试了试让电脑连接手机热点,竟然一次成功为mxnet配置cuda 9.2

P.S.:如果之前的数据集是手动配置的话,更新过后的源代码还会被还原。所以需要重新做一遍配置本地数据集的操作,具体见我的上一篇博客。

3,实验

为mxnet配置cuda 9.2

用书上AlexNet的demo验证,确实在gpu上运行了