diff --git a/install.js b/install.js index ddcfcab..b841e6a 100644 --- a/install.js +++ b/install.js @@ -8,7 +8,7 @@ function install(kernel) } if ([ 'linux', 'win32' ].includes(platform) && gpu === 'nvidia') { - return 'python install.py --onnxruntime cuda-12.2'; + return 'python install.py --onnxruntime cuda-11.8'; } if (gpu === 'amd') { @@ -37,6 +37,17 @@ module.exports = async kernel => message: 'git clone https://github.com/facefusion/facefusion --branch 2.5.2 --single-branch' } }, + { + method: 'shell.run', + params: + { + message: 'conda install cudatoolkit=11.8 cudnn=8.9.2.26 conda-forge::gputil=1.4.0 --yes', + conda: + { + name: 'facefusion' + } + } + }, { method: 'shell.run', params: