0、安裝cuda_11.8,安裝完成後,下載cudnn archive並解壓縮到cuda的安裝目錄。使用visual studio installer安裝C++ build tool和windows SDK(按照安裝的默認選項即可)。
1、在conda新建一個虛擬環境
conda create -n facefusion python=3.10
2、安裝pytorch,讓pytorch使用gpu運作
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
3、新建一個文件夾,在文件夾內獲取facefusion代碼
git clone https://github.com/facefusion/facefusion.git
4、安裝環境依賴
pip install -r requirements.txt
5、安裝ffmpeg
conda install ffmpeg
6、應該可以正常使用gpu運行了
python run.py –execution-providers cuda