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()給出的當前裝置的統計資訊。- 返回型別
