如何使用 gpu in.py 文件 Google 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文件中（在驱动器中）时，我从 shell 使用

!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"
]

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

Answer 1

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

Answer 2

您可以使用@Anwarvic 的建议：

%run file.py

它对我来说非常有效。