解决什么问题
搭建一个文生图环境 + 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