From 1743574eb1ddf1d7c110c33e8223982f0881de17 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 9 Apr 2024 19:06:02 +0200 Subject: [PATCH] Remove custom cuda install --- install.js | 25 +++++-------------------- 1 file changed, 5 insertions(+), 20 deletions(-) diff --git a/install.js b/install.js index 7829839..f25156a 100644 --- a/install.js +++ b/install.js @@ -4,24 +4,24 @@ function install(kernel) if (platform === 'darwin') { - return 'python install.py --onnxruntime default --skip-venv'; + return 'python install.py --onnxruntime default'; } if ([ 'linux', 'win32' ].includes(platform) && gpu === 'nvidia') { - return 'python install.py --onnxruntime cuda-11.8 --skip-venv'; + return 'python install.py --onnxruntime cuda-12.2'; } if (gpu === 'amd') { if (platform === 'linux') { - return 'python install.py --onnxruntime rocm-5.4.2 --skip-venv'; + return 'python install.py --onnxruntime rocm-5.4.2'; } if (platform === 'win32') { - return 'python install.py --onnxruntime directml --skip-venv'; + return 'python install.py --onnxruntime directml'; } } - return 'python install.py --onnxruntime default --skip-venv'; + return 'python install.py --onnxruntime default'; } module.exports = async kernel => @@ -30,21 +30,6 @@ module.exports = async kernel => { run: [ - { - when: '{{ gpu === "nvidia" }}', - method: 'shell.run', - params: - { - message: - [ - 'conda install cudatoolkit=11.8 cudnn=8.9.2.26 -y' - ], - conda: - { - name: 'facefusion-pinokio' - } - } - }, { method: 'shell.run', params: