- 博客(3)
- 资源 (7)
- 收藏
- 关注
原创 MXNet多机分布式配置
mxnet多机分布式配置将一台机器上的mxnet拷到另外两台机器上scp -r /home/hzz/mxnet hzz@9.91.21.37:/home/hzz/mxnet如果已经在自己的环境安装好mxnet或者不想新建一个Python环境,可以忽略这几步。建立新的Python环境对每台机器,用anacodna的conda建立新的Python环境,在这个环境安装mxnet...
2018-06-27 14:24:26 3206
原创 MXNet下,随机生成图片Iter
为了测试有时需要随机生成图片文件,为了自己predict需要去掉了label。 MXNet的module比较麻烦,一定需要使用DataIter,于是自己写了测试使用的DataIterclass RandomDataIter(mx.io.DataIter): def __init__(self, batch_size, max_iter=1000, dtype=np.float32, ...
2018-06-22 15:41:23 428
原创 Python查看GPU已使用的显存
pip安装pynvmlpip install nvidia-ml-py3如果是Python2,则pip nvidia-ml-py2之后可以在Python3 下使用了import pynvmlpynvml.nvmlInit()# 这里的0是GPU idhandle = pynvml.nvmlDeviceGetHandleByIndex(0)meminfo = pynvml...
2018-06-22 15:36:40 27418 2
mobilenetv2_0.75.pth
2019-12-23
darknetcoco.pth
2019-12-23
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人