2023年5月4日
Python 利用 Pytorch 获取当前GPU信息
摘要： import torch def get_gpu_info(): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") if device.type == "cuda": # 获取当前GPU名字 gpu_name 阅读全文
posted @ 2023-05-04 21:18 ~逍遥子~ 阅读(856) 评论(0) 推荐(0) 编辑