Jupyter Notebook GPU memory 训练后发布 model

Question

在使用 Jupyter notebook 完成深度学习 model 训练后，我们如何清理 GPU memory。 问题是，无论我坚持使用什么框架（tensorflow，pytorch），存储在 GPU 中的 memory 都不会被释放，除非我手动终止进程或终止 Z50484C19F1AFDAF3841A0D821ED3 并重新启动。 您知道我们如何通过自动化步骤来解决这个问题吗？

Answer 1

我发现的唯一解决方法是使用线程。 使用子流程执行培训。

一个例子：

def Training(arguments):
    ....
    ....
    return model   

if __name__=='__main__':
Subprocess = Process(# The complete function defined above
                     target = Training,
                     # Pass the arguments defined in the complete function 
                     # above.
                     # Note - The comma after the arguments : In order for 
                     # Python
                     # to understand this is a tuple
                     args = (arguments, ))

# Starting the defined subprocess
Subprocess.start()
# Wait for the subprocess to get completed
Subprocess.join()