如何在kaggle免费使用GPU部署Stable Diffusion?

一、Stable Diffusion

目前AI绘画主要是用Midjourney和Stable Diffusion两款工具。 Midjourney是付费的,而Stable Diffusion是免费开源的。 Stable Diffusion是基于PyTorch的模型,它需要以下条件来部署和使用:

  1. PyTorch: 必须安装PyTorch,最好是最新版本的PyTorch。您可以通过官方网站下载和安装PyTorch。
  2. CUDA: Stable Diffusion需要GPU支持,所以您需要一个支持CUDA的GPU和安装了适当版本的CUDA库。
  3. Python: Stable Diffusion是用Python编写的,因此您需要安装Python环境(最好是Python 3.6或以上版本)。
  4. 依赖项: Stable Diffusion使用了多个Python库,如numpy、scipy、torchvision等等。您需要安装这些依赖项才能运行模型。
  5. 内存和磁盘空间:部署Stable Diffusion需要一定的内存和磁盘空间。特别是,在运行模型时,可能需要较大的GPU内存。
  6. 网络连接:在使用Stable Diffusion时,需要一个稳定的网络连接,以便在需要时加载和使用模型。

如果没有很好的硬件支撑,还可以利用可以提供GPU的平台,然后将Stable Diffusion部署在上面。
今天要介绍的是kaggle,它提供免费的GPU服务,我们可以利用它来搭建Stable Diffusion。

二、kaggle介绍

Kaggle是一个流行的数据科学和机器学习竞赛平台,也是数据科学家、机器学习工程师和研究人员分享、发现和交流知识的社区。Kaggle提供许多数据集和竞赛,鼓励用户分享并提交他们的解决方案。该平台还提供了一个云端的Jupyter Notebook环境,让用户可以轻松地共享、重复和修改其他人的代码,并与广泛的社区进行交流。此外,Kaggle还提供对GPU和TPU的免费访问,以加速数据科学项目的执行速度。Kaggle是一个非常有用和便捷的平台,为对数据科学和机器学习感兴趣的人提供了一个很好的交流和实践的机会。

官网:kaggle.com

三、注册流程:

点击右上角Register进行注册。注册使用过程需要用到科学上网。

可以用gmail邮箱进行注册。
注册成功之后,会进入kaggle的一个个人主页。点击左下角的zh-stable-diffusion-webui-kaggle的选项可以进入到stable diffusion wenbui的部署页面。

手机验证:在Notebook options下找到AddTags,再点击Get Phone verified,会弹出手机验证界面,只要找到CN +86,然后输入手机号即可。

四、选项配置:

验证完成之后需要做以下配置选择。

INTERNET:选择on

ACCELERATOR:选择GPU T4 x2

五、启动运行:

先点击电源键进行开机,然后点击运行,便会开始启动部署。

整个启动过程需要几分钟,待提示Running on public URL时,说明已经部署完成。

六、使用stable-diffusion-webui

直接点击对应的链接就可以打开访问zh-stable-diffusion-webui-kaggle了。输入咒语即可以绘图了。

而且用手机也可以直接打开:

七、关闭:

一周有30小时的免费使用时长,具体使用时长可以可以实时看到。如果不用时可以选择关闭,关闭后不再计时。

THE END