AIGC：python 文生图代码（python + stable-diffusion+ cuda)

解决什么问题

搭建一个文生图环境 + python文生图代码，通过输入prompt,生成图片

解决方法

文生图的代码比较简易，主要是搭建环境+下载文生图大模型

步骤一：
创建虚拟环境(用python 3.10版本，起得名字最好可以一看就知道Python的版本）
conda create -n text2img3.10 python==3.10.4

步骤二：
搭建torch环境
直接用pip install 安装网络上的容易出现问题
我们先下载到本地，然后再用pip install 安装
先看下电脑cuda的版本：nvidia-smi

可以看出版本号是12.3，我们可以选择torch 的cu12.3或者低于12.3都可以
A. 下载torch网址：https://download.pytorch.org/whl/cu116/torch
根据创建环境3.10和win还是linux来选择torch版本

我的是python 3.10.4 + win，所以选择的是画红色线的那个版本，如果你的是Linux环境，请选择画红线的上面一个版本

B. 下载torchvision
网址：https://download.pytorch.or

文章来源于互联网:AIGC：python 文生图代码（python + stable-diffusion+ cuda)

相关推荐: AIGC 实践全攻略：开启你的智能创作之旅！

AIGC 实践全攻略：开启你的智能创作之旅！ 随着人工智能生成内容（AI-Generated Content, AIGC）技术的迅速发展，我们正迎来一个全新的创作时代。无论是写作、绘画，还是视频剪辑，AIGC 都正在改变内容创作者的工作方式，让更多人拥有创造的…