完整代码如下:

pytorch程序异常后删除占用的显存操作