memory_cached has been renamed to torch.cuda.memory_reserved . So use memory_cached for older versions. Output: Using device: cuda Tesla K80 Memory Usage: ...
確定! 回上一頁