当前位置: 首页 > 工具软件 > Apache MXNet > 使用案例 >

Apache MXNet环境搭建

邬飞捷
2023-12-01

最近想学习一下深度学习相关的知识,在搭建环境的时候遇到了好几个问题,查了下这些问题,发现蛮多解决方案都不行的,很多人都卡在了环境搭建这一步,所以记录一下可能能帮到卡住的同学。
首先打开https://zh-v2.d2l.ai/chapter_installation/index.html 这个网站,前面几步照着做就行,然后来到配置GPU版本的mxnet,可能很多人都卡在了这一步,查这个问题的话会发现很多直接给你命令的解决放式,我试过都不行,报错提示没有匹配的mxnet版本,其实原因是后面才开始学习的人打开CUDA
Tootkit默认提供的版本是11.x的,打开https://dist.mxnet.io/python/看了一下里面,确实没有对应版本mxnet,windows下的只有10.2的,所以下载10.2的cuda工具,然后用官网给的指令就能安装,国内最好加个映像会快一点,像这样:

pip install mxnet-cu102==2.0.0b20201108 -f https://dist.mxnet.io/python -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

接着如果你没装vs2019可能还会有报错,据说是没有c++生成工具导致的,所以搭建环境之前要先安装一下2019。
我遇到的问题就这两个,尤其第一个感觉挺多人问的,但也没有好的答案,希望能帮到卡在环境搭建的同学。

 类似资料: