查看GPU是否在colab中
import tensorflow as tf
tf.test.gpu_device_name()
如果结果为空,则不能使用GPU,如果结果为/device:GPU:0
使用!/opt/bin/nvidia-smi查看显存情况
查看显卡内存使用上限
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
查看GPU是否在colab中
import tensorflow as tf
tf.test.gpu_device_name()
如果结果为空,则不能使用GPU,如果结果为/device:GPU:0
使用!/opt/bin/nvidia-smi查看显存情况
查看显卡内存使用上限
from tensorflow.python.client import device_lib
device_lib.list_local_devices()