深度学习
派大星的好朋友叮当猫
这个作者很懒,什么都没留下…
展开
-
【GPU】关于GPU的基本操作(一)
目录1.检测NVIDIA驱动是否成功安装2.查看集显和独显1.检测NVIDIA驱动是否成功安装英伟达系统管理接口( 简称 nvidia-smi)nvidia-smi#帮助管理和监控NVIDIA GPU设备执行该命令将会打印出当前系统安装的NVIDIA驱动信息显示可以看到显卡的型号是 Tesla K802.查看集显和独显lspci | grep VGA #显示集成显卡lspci | grep NVIDIA#显示NVIDIA显卡两者都能搜索到则说明正常...原创 2020-05-25 18:33:30 · 690 阅读 · 0 评论 -
【深度学习】数据分割
在训练模型前,要把数据集进行分割,按照一定比例分成训练集、测试集、验证集,在有些时候也可以忽略测试集,笔者这里是分了测试集(训练集:测试集:验证集=8:1:1),废话不说,直接上代码# h_split.pyimport osfrom shutil import copyimport randomdef mkfile(file): if not os.path.exists(...原创 2020-04-01 09:31:11 · 1280 阅读 · 6 评论