GTX 1080 + macOS10.13.2 + MXNet

GTX 1080 + macOS10.13.2 + MXNet,第1张

最初打算买1080 Ti,发现1080 Ti+拓展坞的组腔态闷合比Gigabyte AORUS GTX 1080 Gaming Box贵很多,犹豫再三,还是购买了1080,妥妥地够用。

Gigabyte AORUS GTX 1080官方驱动不支持macOS。我的系统是macOS10.13.4,查过很多资料后,发现解决方案比较伍弯麻烦,需要使用macOS 10.13.3的kext,如下图:

最终还是将系统从10.13.4降到了10.13.2。严格按照 步骤 安装:

安装成功后在Graphics/Displays和NVIDIA Driver Manager可查到闭御外置显卡的信息:

目前无法通过pip安装GPU版本的MXNet,只能通过 源码 )安装:

安装好CUDA9.1后,测试GPU:

安装好cuDNN7.0.5,构建MXNet,最后采用基于MXNet的测试用例验证一下包:

安装过程中需要注意的是:

用install_name_tool将lib指向修改为正确的即可:

Win10下安装CUDA10.2版本Mxnet,咐搭当前只找到 mxnet_cu102-1.7.0-py2.py3-none-win_amd64.whl 。

安装后安装gluoncv,通过 git clone https://github.com/dmlc/gluon-cv 获得到了源码,经验瞎简滑如磨腊下:

安装mxnet

Cpu版:pip install mxnet

Gpu版:pip install mxnet-cu80

若cuda版本为cuda-9.0,改为pip install mxnet-90

pip或apt安装sklearn,easydict

制作数据

图片分类并放入不同文件夹如纯

运行python im2rec.py train --list ./可生成.list文件,包含图片列表

运行python im2rec.py train ./ train.rec和train.idx训练文件

--train-ratio 0.9生成验证数据集.bin文件参数为训练数据集和验证数据集之比

--resize 128 128指定生成数据集的图片大小

参考 https://github.com/apache/incubator-mxnet

将生成的.rec,.idx,.bin(非必须)文件放入datasets/faces_emore中

新建property文本,写入图片数量,图片长宽 example : 86545 128 128

例:

python -u  train.py --network m1 --loss softmax --dataset emore,1

使用softma、nosoftmax、arcface或cosface训练完成后,使用生成模型运行三元组损失局橡雀训练

例:

Python -u train.py --network m1 -loss triplet --lr 0.005 --pretrained ./models/m1-softmax-emore

参数说明

--dataset训练集位置,具体位置查看config.py 108至120行

--network网络模型 候选参数 : r100 r100fc r50 r50v1(基于resnet) d169 d201(基于densenet) y1 y2(基于mobilefacenet) m1 m0.5(基于mobilenet) mnas mnas05 mnas025(基于桐早mnasnet)

--loss损失函数 候选参数 :softmax(标准损失函数) nsoftmax (组合损失函数)arcface cosface combined triplet(三元组损失) atriplet

--ckpt模型存储时间。0: 放弃存储 1:必要时存储(验证集准确率达标时,若无验证集则不存储 3:总是存储)

--lr学习率

--lr-steps学习率改变方法 例:’10000,20000,2200000’即达到图片数量时学习率*0.1

--per-batch-size每次的训练的数量 数量越少,占用显卡内存越少

参考

https://github.com/deepinsight/insightface/tree/master/recognition


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/tougao/12483430.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023-05-25
下一篇 2023-05-25

发表评论

登录后才能评论

评论列表(0条)

保存