# 如果点用了cuda,则将gpu的相关信息打印出来。
# show = True表示打印, show = False表示不打印
if device.type == "cuda":
pmr.get_gpu_prop(show=True)
【笔记】调用gpu并显示gpu的基本信息
最新推荐文章于 2023-12-21 20:52:02 发布
# 如果点用了cuda,则将gpu的相关信息打印出来。
# show = True表示打印, show = False表示不打印
if device.type == "cuda":
pmr.get_gpu_prop(show=True)