Colab的使用

Colab 是谷歌提供的在线 NoteBook 平台,支持 GPU, TPU 加速,并且免费。

一、相关命令

导入 TensorFlow 版本

1
2
3
4
# 选择导入 TensorFlow 版本
%tensorflow_version 2.x
import tensorflow as tf
print(tf.__version__)

查看 GPU 信息

1
2
3
4
5
6
7
8
# 查看 GPU 信息
gpu_info = !nvidia-smi
gpu_info = '\n'.join(gpu_info)
if gpu_info.find('failed') >= 0:
print('Select the Runtime → "Change runtime type" menu to enable a GPU accelerator, ')
print('and then re-execute this cell.')
else:
print(gpu_info)

查看CPU信息

!cat /proc/cpuinfo

查看内存

!cat /proc/meminfo

挂载 Google Drive

1
2
from google.colab import drive
drive.mount('/content/gdrive')

二、Shell

在 NoteBook 中代码编辑中加上!

例如:查看当前路径 ! pwd

  • 本文作者: Ashin Wang
  • 本文链接: https://ashin.wang/use-colab/
  • 版权声明: 本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!