Colab 的使用

使用 Colab 的时候，往往需要访问其他文件，比如自己的数据集等，这里记录一种比较简单的方法。

我们先将需要使用的文件夹上传到 Google Drive上，就像下面这样

例如我将 assignment3 整个文件夹上传到 Google Drive 中, 然后使用Colaboratory打开文件夹中的 GANs-PyTorch.ipynb 文件

如果直接运行其中的代码的话，遇到像下面这种读取文件的代码，会产生错误

answers = dict(np.load('gan-checks-tf.npz'))

所以我们要有如下几步设置

挂载谷歌硬盘

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

这里会出现一个链接进行授权，将授权后的代码复制到框中即可

此时可在右边看到文件目录结构:

现在我们还是不能按照相对路径的方式读取文件，因为我们还未修改工作目录

更改工作目录

在 Colab 中可以使用Linux命令，但有点不同。

%pwd

使用上面的命令查看当前的工作目录，一般是/content/gdrive

我们现在需要修改工作目录，通过下面的命令切换工作目录

%cd /content/gdrive/My\ Drive/Colab Notebooks/assignment3
%pwd

• 注意使用%cd来切换, !cd没有用
• 空格用\转义
• 切换到自己的目录就行了，上面只是一个例子

切换工作目录后，就能使用相对路径的方式访问到谷歌硬盘中的文件，上面的代码就不会出错啦.

使用GPU

选择代码执行程序->更改运行时类型->GPU 即可

查看是否为GPU设备

import tensorflow as tf
tf.test.gpu_device_name() # 查看设备名字

查看显卡情况

!/opt/bin/nvidia-smi # 查看显卡情况