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