难道你不想用AI资源生成个女朋友吗

457 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 5 天,点击查看活动详情

AI火了

最近什么最火?

ChatGPT,OpenAI,还有AI绘画领域的Stable Diffusion,文生图(txt2img),图生图(img2img)。

你最近也概率看到过类似下面这样的图:

00001.jpg

你是不是也想自己去按照想象生成这样的图呢?

安装执行跑

废话不说,上代码:

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
# 下面这一句可能会引发报错,主要是看电脑上是否安装了合适的显卡驱动/cuda/pytorch/torchvision
pipe = pipe.to("cuda")
prompt = "a photograph of an astronaut riding a horse"
image = pipe(prompt).images[0]
img.save(f"astronaut_rides_horse.png")

其中最麻烦的是pipe = pipe.to("cuda")这一句,需要根据你的显卡,去安装合适的驱动,然后安装合适的cuda,再安装pytorch, torchvision。。。

是不是很简单,就好像买个劳斯莱斯一样!

不错,程序简单,但是本地运行这样的AI模型一般需要不错的显卡来支撑,例如我本地电脑上安装的是NVIDIA GeForce GT 730显卡,就是死活跑不起来,默认安装pytorch会安装成CPU的版本,指定GPU安装,又会提示各种python环境不匹配,最后的最后也没有跑起来,只勉强使用CPU跑了两次,而且,跑的时候,电脑什么也干不了,因为执行这个几乎占用了你CPU的全部计算能力。

那么我们普通人,不能向老婆/女朋友申请经费的人,怎么能愉快地发挥想象力呢?

答案就是借助于大厂平台或者免费的App。

那些慷慨的大厂

Google Colab

在网上偶然发现了一个宝藏资源,Google Colab,操作界面大致如图:

image.png

在里面可以很方便的记录笔记,执行命令/代码/Python文件。

Getting started with Diffusers,这是一个使用Google Colab可以打开的一个HuggingFace教程笔记,笔记可以放置在Github或者Google Drive上。

进阶绘图

只使用代码绘图多少原始了一些,那么是否有图形界面呢?

这是一个图形化绘图笔记 ,执行后跑起来的界面如下所示:

image.png

具体怎么操作,你可以去B站或者Youtube上搜AI绘图看看教程,就不一一列举了。

其他类似的深度学习平台有:Kaggle, 百度的飞桨(亲测不能跑pytorch),阿里的天池,DataCastle,Datafountain,科赛-Kesci,biendata,华为云的ModelArt等等

用大厂免费的显卡CPU和硬盘不香吗?