【发布时间】:2018-07-08 13:19:07
【问题描述】:
Here 描述了如何在 google-colaboratory 中使用 gpu:
只需在笔记本设置的加速器下拉菜单中选择“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。似乎描述不完整。有什么想法需要做什么吗?
【问题讨论】: