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