基于pytorch的python程序。 - Python论坛

编程论坛 → 开发语言 → 『 Python论坛』 → 基于pytorch的python程序。

欢迎加入我们，一同切磋技术

共有 999 人关注过本帖

标题：基于pytorch的python程序。

只看楼主加入收藏

abc3270560

等　级：新手上路
帖　子：1
专家分：0
注　册：2019-4-18

问题点数：0 回复次数：0

基于pytorch的python程序。

小白最近在跑一个关于GAN的代码，但在这个代码中调用的一个程序中出现了一些问题。总的表现为超出显存。但仔细排查后发现程序每次循环的时候都把它自己定义的模型给送到GPU 了，导致显存激增，且循环完毕也不释放。想求教各路大神怎么解决这个问题。改网络结构的话，以我目前的水平是改不了。希望大神们可以帮忙！

搜索更多相关主题的帖子: python　代码　循环　释放　结构　

2019-04-18 22:07

快速回复：基于pytorch的python程序。