使用 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 # 查看显卡情况