使用硬件加速时的GPU内存

时间:2018-02-28 01:03:46

标签: google-colaboratory

我一直在colab上遇到问题(特别是在使用PyTorch时),使用reject()方法的中断内核在重新启动时会耗尽内存。

一位同事建议在colab平台上的不同用户之间共享GPU内存。这似乎是一个非常糟糕的主意,但可能是这个问题的一个答案。任何人都可以确认硬件加速器是专门针对特定用户在colab上的会话吗?

由于

1 个答案:

答案 0 :(得分:1)

确认每个硬件加速器都分配给一个用户,并且每个笔记本电脑都有自己的硬件分配。 (前者自发布以来就适用,而后者自最近以来就适用)。

该问题使用表达式“用户的会话”,但未定义它。为了具体起见,上述排他性适用于(用户,笔记本)对。 (并且不包含浏览器标签)。