摘要:
📘 查看服务器显卡使用情况 一、命令行运行python程序时 首先查看哪些GPU空闲,nvidia-smi显示当前GPU使用情况 nvidia-smi 如下图所示:服务器中的两个显卡,编号为0、1 . 都被同一个进程 PID 3016 占用 图示基础信息 GPU:GPU 编号; Name:GPU 阅读全文
摘要:
inputs = {"input_ids": batch[0], "attention_mask": batch[1], "labels": batch[3]} if args.model_type != "distilbert": # XLM and RoBERTa don"t use segme 阅读全文