深度学习
TccccD
这个作者很懒,什么都没留下…
展开
-
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 · 27425 阅读 · 2 评论 -
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 · 430 阅读 · 0 评论 -
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 · 3207 阅读 · 0 评论