您即将离开知乎,请注意您的账号和财产安全。
https://stackoverflow.com/questions/59129812/how-to-avoid-cuda-out-of-memory-in-pytorch