使用Google Colab运行Tensorflow 2 GPU

使用Google Colab运行Tensorflow 2 GPU

在菜单栏中选择“修改”--“笔记本设置” ，硬件加速选择GPU（Tesla K80 GPU）

卸载原有tensoflow版本：

!pip uninstall tensoflow

安装tensorflow 2 的GPU版本

!pip install tensorflow-gpu 

然后再安装上一些依赖包

!pip install tf-nightly

查看此时是否以GPU运行：

import tensorflow as tf
tf.test.is_gpu_available()
tf.test.gpu_device_name()

此时输出True则可以使用GPU，设备输出'/device:GPU:0'

 

 

注意：此方法只能在这个.ipynb文件下运行，新建另一个.ipynb文件需要重新配置