使用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)

运行正常可以看到如下结果:
Colab使用教程

否则看到如下结果:
Colab使用教程

或者使用命令

!nvidia-smi

可以看到以下结果:
Colab使用教程

否则看到如下结果:
Colab使用教程

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

切换文件夹

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

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

参考

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

相关文章:

  • 2022-01-20
  • 2021-12-01
  • 2021-11-27
  • 2021-07-24
  • 2021-12-30
  • 2021-11-19
  • 2021-10-14
  • 2021-12-30
猜你喜欢
  • 2021-11-18
  • 2021-12-15
  • 2022-12-23
  • 2021-11-19
  • 2021-12-04
  • 2021-06-19
  • 2021-10-16
相关资源
相似解决方案