Google colab的GPU是否可以与Tensorflow 1.4.0版本一起使用吗?

时间:2020-07-13 08:51:23

标签: python tensorflow

当我在Google colab笔记本中检查tensorflow的版本时,我猜它是预安装的2.2.0版本。我从笔记本设置中启用了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))

它提供了所需的输出,但我的代码需要tensorflow 1.4.0,因此我使用!pip install tensorflow == 1.4.0安装了它。它安装成功,但是当我重新启动运行时并检查GPU时,它给出了以下错误

SystemError:找不到GPU设备

我错过了什么

0 个答案:

没有答案