当我运行一段代码的时候出现一开始显存占用没有满,随着训练的step增多显存逐渐增大。我尝试了https://blog.youkuaiyun.com/qq_36401512/article/details/96163940所有方法发现都不行。直到收到一位高人指点,将原先的torch1.7版本换为torch1.6问题就解决了。
当我运行一段代码的时候出现一开始显存占用没有满,随着训练的step增多显存逐渐增大。我尝试了https://blog.youkuaiyun.com/qq_36401512/article/details/96163940所有方法发现都不行。直到收到一位高人指点,将原先的torch1.7版本换为torch1.6问题就解决了。