您即将离开知乎，请注意您的账号和财产安全。
https://pytorch.org/docs/main/torch_cuda_memory.html#understanding-cuda-memory-usage
