我正在浏览Artistic Styling transfer tutorial in CNTK。我可以通过一个图像，然后必须重新启动jupyter笔记本，以避免耗尽GPU内存来设置附加图像的样式。错误代码是打击。我想把最后2个代码块放在一个for循环中，该循环读入图像文件夹并为每个图像创建一个新的样式图像。我想我要做的就是转储GPU内存。如果这是正确的，我该怎么做？
RuntimeError: CUDA failure 2: out of memory ; GPU=0 ; hostname=JOHN ; expr=cudaMalloc((void**) &deviceBufferPtr, sizeof(AllocatedElemType) * numElements)
答案 0 :(得分：0)
在处理第二张图片之前，您可以尝试拨打电话
del content_activations
del loss
import gc
gc.collect()
这些是需要重新定义的对象，这样就可以在构建下一个content_activations和loss之前清理它们（从而释放GPU内存）。
答案 1 :(得分：-1)