如何在.py文件中使用 gpu 谷歌 Colab?

Question

我在GC中有一个笔记本，配置了GPU计算。当我在此笔记本中运行时：

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

我可以在设备中看到 GPU：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 2881683908768157563
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 14433918882325490908
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 8857856280193037152
physical_device_desc: "device: XLA_GPU device"
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 15701463552
locality {
bus_id: 1
links {
}
}
incarnation: 13142570581108506915
physical_device_desc: "device: 0, name: Tesla P100-PCIE-16GB, pci bus id: 0000:00:04.0, compute capability: 6.0"
]

但是当我把这段代码(没有别的(放在.py文件(在驱动器中(中时，我使用

!python3 file.py

我在设备中看不到 GPU：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 5088676628756383267
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 550355900248427842
physical_device_desc: "device: XLA_CPU device"
]

如何解决？ 即如何在从笔记本终端运行的.py文件中使用 GPU 计算？

Answer 1

您可以使用@Anwarvic的建议：

%run file.py

它对我来说非常有效。

Answer 2

我实际上不确定为什么会发生这种情况，但我仍然认为您正在运行的文件不需要使用 GPU，因此没有使用 GPU。如果我错了，请纠正我。