在谷歌C​​olaboratory,有没有办法支付更大的GPU?

时间:2018-02-20 15:49:32

标签: gpu google-colaboratory

我测试了对Colaboratory的GPU支持。 GPU的内存有限。

name: "/device:CPU:0" 
device_type: "CPU" 
memory_limit: 268435456 locality { } 
incarnation: 14648174833476954761, 
name: "/device:GPU:0" 
device_type: "GPU" 
memory_limit: 356515840 
locality { bus_id: 1 } 
incarnation: 11566567776783368174 
physical_device_desc: "device: 0, 
name: Tesla K80, 
pci bus id: 0000:00:04.0, 
compute capability: 3.7"]

GPU内存的大小是356MB,我想知道是否有办法获得付费版本以增加GPU内存?也许这已经是付费谷歌产品的一部分,有谁知道该产品是什么?

3 个答案:

答案 0 :(得分:4)

未共享Colab中的GPU。每个用户都可以完全访问具有12G内存的k80。

看到356MB的可用GPU内存几乎总是表明您创建的TensorFlow会话没有使用allow_growth = True选项。看到: https://www.tensorflow.org/guide/using_gpu#allowing_gpu_memory_growth

如果您未设置此选项,则默认情况下,TensorFlow将在创建会话时保留几乎所有GPU内存。

好消息:从本周开始,Colab现在默认设置此选项,因此当您在Colab上使用多个笔记本时,您应该看到增长率大大降低。而且,您还可以通过在运行时菜单中选择“管理会话”来检查每个笔记本的GPU内存使用情况。

enter image description here

选择后,您将看到一个对话框,列出所有笔记本和每个笔记本所消耗的GPU内存。要释放内存,您也可以从此对话框终止运行时。

enter image description here

答案 1 :(得分:3)

Google推出了适用于colab的专业版,该版本可以使用更好的GPU-https://colab.research.google.com/signup

答案 2 :(得分:1)

目前无法通过Colab请求更大的GPU。