torch.cuda.device_memory_used¶
- torch.cuda.device_memory_used(device=None)[source][source]¶
以字节为单位返回由 nvidia-smi 或 amd-smi 提供的已使用的全局（设备）内存。
- 参数
device (torch.device or int, optional) – 选定的设备。如果
device为None(默认)，则返回由current_device()给出的当前设备的统计信息。- 返回类型
