Google Colab 部署自己的机器学习项目

1. 介绍

Google Colab提供的是免费Tesla K80 GPU,可以玩Keras、Tensorflow、PyTorch或者Mxnet等。Tesla K80还是一块比较高性能的GPU。主要是免费,免费,免费 !!! 好像这次真的薅资本主义羊毛了,个人试了几天,感觉还不错。刚入手的时候稍微有点麻烦,大部分的网上教程都是在 Colab 上手动写代码玩玩儿,但是我是自己在本地写好了一个小的项目,然后相互之间 import 有点麻烦,最后自己摸索总结出了一套最简单的上传自己项目,用 colab 跑出结果的方法。

2.准备工作

  • 谷歌一系列,所以先得出去
  • 注册一个谷歌账号
  • 用自己的谷歌账号使用 google driver
  • 自己的项目在本地调试通过,放在一个文件夹里面(包含数据集,最好是英文名字)

3.步骤

3.1 进入 google driver:https://drive.google.com

3.2 右键上传项目所在的文件夹:

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190703225625905.png在这里插入图片描述
我这里简单地上传了一个名为 test 的文件夹,里面只有一个 main.py 的文件,代码也很简单,但是可以简单测试 Pytorch 的使用。

3.3 进入 colab:点击左上角 “新建” --> “更多” --> “Colaboratory”

在这里插入图片描述
在这里插入图片描述

3.4 “Edit” --> “Notebook settings” -->

在这里插入图片描述
这里我选择的 Python3 (根据自己代码依赖的环境) 和 GPU,如果要用 GPU 这一步必须调整好。每次调整这里的设置,后面的步骤都要再次进行。所以推荐这里设置好
在这里插入图片描述

3.5 代码段里面输入

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

然后点击左侧的运行图标,或者 Ctrl + Enter 运行。这时候会出现
在这里插入图片描述
点击 URL 进入,授权 Google 用户,然后复制授权码粘贴到输入框,Enter。这个时候回显示挂载成功
在这里插入图片描述
这里可以输入

!ls

在这里插入图片描述
可以看到此目录下文件。

3.6 代码段里面输入

import os
os.chdir(“drive/My Drive/test”)

最后一个 test 是其实是自己的项目名。这个时候没有任何提示信息,因为是 Linux 内核,所以按照 Linux 的思想:没有消息就是最好的消息。说明一切顺利。
这里可以输入

!ls
在这里插入图片描述

可以看到我们写的 main.py 文件。

3.7 代码段里面输入

!python main.py
在这里插入图片描述

成功搞定~

4 注意事项

  1. 虽然是 Linux 内核,但是每次类似于 Linux 的语句之前要多一个 “!” 。比如步骤 3.6, 3.7
  2. 步骤 3.6 第二个语句最后一个路径名称根据自己上传到 Google Driver 的文件夹的名字而定,我上传的是 “test ”,所以最后一个路径是“test”。
  • 15
    点赞
  • 35
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
Colab是一个基于云端的Jupyter笔记本环境,可以免费使用GPU和TPU资源。DDPM(Diffusion Probabilistic Models)是一种生成模型,用于生成高质量的图像。在Colab部署DDPM扩散模型可以让你在云端快速搭建和训练模型。 以下是Colab部署DDPM扩散模型的步骤: 1. 打开Colab网站(https://colab.research.google.com/)并登录你的Google账号。 2. 创建一个新的笔记本或者打开一个已有的笔记本。 3. 在笔记本中导入所需的库和模型。你可以使用PyTorch或TensorFlow等深度学习框架来实现DDPM模型。 4. 准备数据集。根据你的需求,可以选择使用已有的数据集或者自己准备数据集。 5. 定义DDPM模型的结构和参数。根据DDPM的论文或者相关资料,实现DDPM模型的网络结构和训练参数。 6. 加载数据集并进行预处理。根据你的数据集格式和要求,加载数据集并进行必要的预处理,例如图像大小调整、归一化等。 7. 定义损失函数和优化器。根据DDPM模型的训练目标,选择合适的损失函数和优化器。 8. 进行模型训练。使用加载的数据集和定义的损失函数、优化器,进行模型的训练。可以设置合适的训练轮数和批次大小。 9. 保存训练好的模型。在训练完成后,保存模型以备后续使用。 10. 进行模型评估和生成。使用保存的模型对新数据进行评估和生成,观察生成效果并进行调整。 以上是在Colab部署DDPM扩散模型的一般步骤。具体实现过程中,可能还需要根据你的具体需求进行一些调整和优化。希望对你有所帮助!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值