先决条件-Git下载链接:https://git-scm.com/downloads/win
教程一:自行下载搭建
大家好,今天我将分享如何在Windows系统中制作AI项目的一键整合包,以著名的换脸项目FaceFusion为例。
所谓整合包,其实就是将程序和运行环境一起打包,这样有利于项目的迁移和传播。我们将依托Anaconda软件来完成这个任务。
首先,下载最新的Anaconda软件。可以直接去Anaconda官网下载。双击安装,这里我已经安装过了,就不再演示安装过程。
随后,打开终端,输入命令conda create,创建一个新的虚拟环境。我们使用Python 3.11,因为这个版本性能有优化,又不会太新导致依赖跟不上版本。命令如下:
conda create -n myenv python=3.11
激活虚拟环境:
conda activate myenv
接着,安装所需的库。我们可以看到最新的支持版本是2.6,最老的是11.3。这里我们选择中间的11.8版本。输入以下命令安装CUDA:
conda install -c nvidia/label/cuda-11.8.0 cuda
# 具体指令需要参考cuda官网
然后安装CUDNN:
conda install cudnn
# 具体指令需要参考cuda官网
这样,环境就配置好了。这个环境位于Anaconda安装目录的envs目录下。
随后,进入FaceFusion官方的仓库,下载最新的代码并解压项目。接着,将刚才创建好的环境拷贝到项目的一个目录中。
用虚拟环境中的Python来安装基础依赖:(也可以将requirements打包好,以避免网络问题:下载离线包)
pip install -r requirements.txt
随后安装Torch三件套:
pip install torch torchvision torchaudio
# 具体指令需要参考torch官网
最后,安装ONNX Runtime GPU:
pip install onnxruntime-gpu
# 具体指令需要参考官网
编写一个脚本,用于检测环境中软件的版本。可以看到环境检测正常。
接着编写脚本,检测ONNX Runtime的后端支持。可以看到也没有问题。
由于项目依赖SPG,这里将SPG软件的目录直接拷贝到虚拟环境目录里。
最后,编写一个批处理文件,用来启动项目。首先是声明Python 3.11的环境变量,然后声明SPG的环境变量,最后是项目执行的Python脚本入口。
当需要对env中虚拟环境进行调整时
CHCP 65001
echo 激活虚拟环境
call activate .\env
start
首次运行会下载模型,模型存放在项目的Assets目录中。首先测试CPU,后端只要不报错就没问题。接着测试CUDA和TensorRT,测试面部超分。
教程二:克隆仓库
CHCP 65001
conda create -p .\env python=3.8
pause
# 更新代码和依赖
@echo off
CHCP 65001
set /p github_address=请输入要更新的GitHub仓库地址:
git clone %github_address% temp
ROBOCOPY temp . /MOV /E
rmdir /S /Q temp
echo 源码更新完成!
env\python.exe -m pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
echo 依赖更新完成!
pause
# 只更新代码
@echo off
CHCP 65001
set /p github_address=请输入要更新的GitHub仓库地址:
git clone %github_address% temp
ROBOCOPY temp . /MOV /E
rmdir /S /Q temp
echo 源码更新完成!
pause
# 只更新依赖
@echo off
CHCP 65001
env\Scripts\pip.exe install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
echo 依赖更新完成!
pause
# 源地址
# 阿里云
-i https://mirrors.aliyun.com/pypi/simple/
# 清华大学
-i https://mirrors.aliyun.com/pypi/simple/
# 中国科技大学
-i https://pypi.mirrors.ustc.edu.cn/simple/
# 豆瓣
-i https://pypi.douban.com/simple/
echo 安装GPU版本torch,详见官网
.\env\Scripts\pip.exe install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
pause
echo 启动程序
.\env\python.exe seg5.py
pause
要安装包:
cd env
python -m pip install
echo 或
.\env\Scripts\pip.exe install