将在tensorflow的能跑通的代码改成pytorch 时,在running时的error 为cuda memory out

试着将tensorflow 的代码转成pytorch 时,cuda memory out ,同是一块显卡.然后调试的时候,只做了3个卷积后就用完了所有显存。
已邀请:

要回复问题请先登录注册