记录MXnet-cu102的安装

#第一次写博客,写的不好不建议,可以把意见提出了,大佬勿喷

1. 查看一下自己的cuda版本 cuda=10.2
而我这边的python版本为3.6

有点难办,如果使用国内源的话,mxnet最新版本为10.1
也就是说10.2的版本是找不到的
在这里插入图片描述
然后我就百度,查博客,看了很多篇文章,发现很多都是10.1,9.0版本的,找不到10.2版本
无奈之下在哪去官网看看
在这里插入图片描述
经过一番查找后,找到了下载地址
在这里插入图片描述

打开网站,往下找找就可以找到使用pip的下载地址
有三个pip
第一个就是卸载CPU版本的mxnet
第二个为windows用户使用的
第三个则是linux或者macOS用户使用的
很明显我们只需要使用第二个
当然,他这里显示的为cu101,而我们只需要改一下就可以正常的运行

pip install mxnet-cu102 -f https://dist.mxnet.io/python

在这里插入图片描述
可以看到找到并能直接下载了
但是一看到这个下载速度我就发愁了
于是,我就保存了这个whl的链接

https://repo.mxnet.io/dist/python/cu102/mxnet_cu102-1.7.0-py2.py3-none-win_amd64.whl

然后去到迅雷哪里下载
在这里插入图片描述
真乃神器也,没有会员也能这么快

这边分享一下
链接:https://pan.baidu.com/s/111auWHDuUceYQn-u6pNREQ
提取码:1eu4
复制这段内容后打开百度网盘手机App,操作更方便哦–来自百度网盘超级会员V2的分享

下载好后,我们就可以开始安装了
在这里插入图片描述
这样子就能安装
在这里插入图片描述
没有报错,安装成功了,接下来开始测试一下

在命令行输入python,进入python环境

import mxnet as mx
a = mx.nd.ones((2, 3), mx.gpu())
b  = a*2+1
b
#这里代码我就不做过多的介绍,想了解的可以百度一下

在这里插入图片描述
可以看到代码成功运行,没有报错
就说明安装成功,并且可以正常使用

好了,这次的安装过程就到此结束

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐