开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 5 天,点击查看活动详情
AI火了
最近什么最火?
ChatGPT,OpenAI,还有AI绘画领域的Stable Diffusion,文生图(txt2img),图生图(img2img)。
你最近也概率看到过类似下面这样的图:
你是不是也想自己去按照想象生成这样的图呢?
安装执行跑
废话不说,上代码:
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,操作界面大致如图:
在里面可以很方便的记录笔记,执行命令/代码/Python文件。
Getting started with Diffusers,这是一个使用Google Colab可以打开的一个HuggingFace教程笔记,笔记可以放置在Github或者Google Drive上。
进阶绘图
只使用代码绘图多少原始了一些,那么是否有图形界面呢?
这是一个图形化绘图笔记 ,执行后跑起来的界面如下所示:
具体怎么操作,你可以去B站或者Youtube上搜AI绘图看看教程,就不一一列举了。
其他类似的深度学习平台有:Kaggle, 百度的飞桨(亲测不能跑pytorch),阿里的天池,DataCastle,Datafountain,科赛-Kesci,biendata,华为云的ModelArt等等
用大厂免费的显卡CPU和硬盘不香吗?