这里描述了如何将 gpu 与 google-colaboratory 一起使用:
只需在笔记本设置的加速器下拉菜单中选择“GPU”(通过编辑菜单或 cmd/ctrl-shift-P 的命令面板)。
但是,当我在笔记本设置中选择 gpu 时,我会弹出一个提示:
分配后端失败 没有可用 GPU 的后端。您想使用没有加速器的运行时吗?
当我运行时:
import tensorflow as tf
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
raise SystemError('GPU device not found')
print('Found GPU at: {}'.format(device_name))
当然,我明白了GPU device not found
。似乎描述不完整。任何想法需要做什么?