把Colab运行后生成的文件保存到Google drive

最新推荐文章于 2025-06-20 09:22:15 发布

原创最新推荐文章于 2025-06-20 09:22:15 发布 · 1.9w 阅读

45 ·

CC 4.0 BY-SA版权

文章标签：

#经验分享

本文介绍如何在Google Colab中将训练好的模型保存至Google云端硬盘，以便下次使用时直接加载，避免重复训练。提供了两种保存方法：一种是简单易行的直接挂载云端硬盘并保存；另一种是通过复杂的挂载代码实现。

问题描述：我是直接新建了一个.ipynb文件后，开始跑代码，断开连接后（比如下工了回宿舍睡觉），下次再打开，生成的模型文件并没有存储，又得重新训练一遍。因此，希望把模型文件存储在drive中，下次就能直接用模型预测，不用再跑一遍代码了。

方法一：

新建一个colab文件：
在这里插入图片描述
点击装载Google云端硬盘：

出现drive文件夹：

运行一个简单的实例：

成功保存。

方法二（复杂）：

运行挂载代码，具体请百度google drive挂载。

!apt-get install -y -qq software-properties-common python-software-properties module-init-tools
!add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null
!apt-get update -qq 2>&1 > /dev/null
!apt-get -y install -qq google-drive-ocamlfuse fuse
from google.colab import auth
auth.authenticate_user()
from oauth2client.client import GoogleCredentials
creds = GoogleCredentials.get_application_default()
import getpass
!google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL
vcode = getpass.getpass()
!echo {vcode} | google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret}

!mkdir -p drive
!google-drive-ocamlfuse drive

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

进入指定目录，因为我的代码是直接新建后跑的，因此与drive都在\content目录下

import os
os.chdir('/content')

我的目录结构如下：

然后执行命令拷贝：

!cp -r checkpoints '/content/drive/MyDrive/ML-2021-spring-yiLee/2020HW9'

这样就把checkpoints这个文件夹保存在drive中，下次打开colab中，文件依然在，可以直接调用。

希望对大家有帮助。