09 2021 档案

gpu轮询显存，自动跑程序脚本
摘要：import os import time gpu_device = 0 cmd = "CUDA_VISIBLE_DEVICES=0 python train.py" def gpu_info(gpu_index): info = os.popen('nvidia-smi|grep %').read 阅读全文
posted @ 2021-09-12 23:46 立冬以东 阅读(252) 评论(0) 推荐(0) 编辑
多线程处理脚本
摘要：在处理数据集的时候，往往是一张一张循环处理，如果数据集太大，处理时间就会很长。我们可以使用多线程处理。 多线程的模板直接用mmcv库里面的一个函数就行了 参考如下： https://github.com/open-mmlab/mmcv/blob/master/docs_zh_CN/understan 阅读全文
posted @ 2021-09-08 19:15 立冬以东 阅读(158) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示