我尝试在 Google Colab 上使用 GPU 运行我的笔记本,但它没有为我提供 GPU,但是当我使用 tensorflow 1.15.0 运行笔记本时,GPU 可用。
tf.test.gpu_device_name()
给出输出 '/device:GPU:0' 对于 tensorflow 1.15.0
但是当我对 tensorflow 2.0.0 执行相同操作时,该函数返回 ''。
请您参考如下方法:
我解决了在 google colab 中安装
!pip install tensorflow-gpu
和
!pip install tf-nightly
所以现在tf.test.gpu_device_name(),输出是/device:GPU:0 但是,TensorFlow 自动将其版本升级到 2.1.0-dev20191120
