首页 > 解决方案 > Adaptis 训练 - GPU 在几次迭代后挂起

问题描述

我在训练 Adaptis ( https://github.com/saic-vul/adaptis/tree/pytorch ) 时遇到问题。当我想训练它时,GPU(RTX 3080)就会停止工作。当它工作时,GPU 需要 95% 的资源,但当它挂起时,它不需要任何资源。

我正在使用 git repo 的 pytorch 分支。

有没有人有同样的问题并且可以帮助我?

计算机:

CPU:锐龙9 5900

内存:32 GB

显卡:RTC 3080

在此处输入图像描述

标签: tensorflowpytorchgpufreezetorch

解决方案


推荐阅读