【发布时间】:2020-07-15 21:19:06
【问题描述】:
我正在通过 pytorch 使用 Google Colab 的 GPU 训练一个 CNN 模型。我的问题是,即使使用相同的代码运行,它有时也会慢三倍(在我的情况下是 30s -> 90s)。
我试过重启运行时(它会清除所有局部变量但保留文件),它不起作用
我看过 this 的帖子,但是,我检查了我的 GPU 状态,它适用于 11.5GB。
有时断开一段时间后会正常,但我仍然想弄清楚可能是什么原因。
【问题讨论】:
-
请分享一个能重现问题的独立笔记本。您是否正在从云端硬盘读取数据文件?通常,驱动器性能差异很大,因此您可以先尝试将文件复制到本地磁盘。
-
@BobSmith 我正在读取本地磁盘中的文件,而且减速问题仅在训练过程中出现,这就是我困惑的原因。
标签: machine-learning pytorch gpu google-colaboratory