Colab使用教程

2019/08/10 01:01
阅读数 1.3W

[toc]

有关链接

使用GPU

以下两种方式都可以:

  1. "修改"->"笔记本设置"->"硬件加速器"选择GPU
  2. "代码执行程序"->"更改运行时类型"->"硬件加速器"选择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')
else: 
  print(device_name)

运行正常可以看到如下结果:

否则看到如下结果:

或者使用命令

!nvidia-smi

可以看到以下结果:

否则看到如下结果:

PS:如果你有其他会话正在运行,可能会看到这样的提示

切换文件夹

不能用Linux下常用的cd命令,要用如下命令

import os
os.chdir("/content/drive/My Drive/your_path")

参考

转载几篇自己使用的时候参考的博客,方便以后查阅。

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部