配置Theano,以便在GPU内存分配失败时不会直接崩溃

时间:2016-08-02 15:31:01

标签: gpu theano

当Theano脚本试图获得比当前可用内存更多的GPU内存时,它会立即崩溃:

F2:    [F`] - [F21] - [F22] - [F23]
      /
M:   [M1] - [M2]
      \    /
       [F']

有没有办法配置Theano,以便在GPU内存分配失败时不直接崩溃?例如,它可以改为每X秒重试一次,并在Y尝试后放弃。

(一个用例:我有几个Theano脚本在同一GPU上运行,并且只有5%的时间使用90%的内存)

0 个答案:

没有答案