MXnet深度学习环境及GluonCV深度学习包的安装
准备使用Gluoncv深度学习包进行图像处理,但是在安装和配置环境时遇到了一些问题,在这里记录一下操作的注意事项
1.最好使用anaconda
在原本的安装过程中,需要安装pytorch框架,gluon官方文档中pip的pytorch是1.6.0版本的,但是我在pip时发现由于python版本太高了(3.11版本),因此在安装时会出错,找不到匹配的版本文件。这时去pytorch官网下载的是最新版本(2.0.1版本),mxnet不支持2.0及以上的pytorch版本.....下载其他版本时发现不支持python3.11版本...使用anaconda配置(python3.6版本)可以省去非常多的功夫,减少很多错误
2.提前安装Microsoft C++ 生成工具
Microsoft C++ 生成工具 - Visual Studio
点击此页面安装,否则MXnet安装会失败
在打开后的页面中选择“使用C++的桌面开发”并安装(可以修改安装包地址)
3.安装CUDA
如果需要GPU进行运算,则需要安装相应版本的CUDA,可以在NAVIDA显卡控制面板——帮助——系统信息——组件 中查看支持的CUDA版本,可以看到最高支持12.1版本的,在这里安装10.2版本的CUDA
前往CUDA安装下载页面
CUDA Toolkit Archive | NVIDIA 开发者
选择10.2版本,具体选项参照如下
下载好安装包后一直确认即可
4.安装MXnet和Gluoncv
Installation — gluoncv 0.11.0 documentation
参照官网的指南进行下载安装,推荐提前安装好opencv,并且从pytorch官网的页面下载相应版本的pytorchPrevious PyTorch Versions | PyTorch
pip安装命令:pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
,
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)