这里描述了如何将gpu与google-colaboratory一起使用:
只需在笔记本设置中的加速器下拉菜单中选择"GPU“(通过编辑菜单或cmd/ctrl 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。看来描述还不完整。有什么想法需要做吗?
https://stackoverflow.com/questions/48508145
复制相似问题