CalvinBackup/facefusion-pinokio
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-pinokio.git synced 2026-07-23 06:20:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simplify install for NVIDIA

Browse Source
This commit is contained in:
henryruhs
2024-09-22 22:48:13 +02:00
parent a0d8d2b124
commit 4db16f0404
1 changed files with 5 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install.js
+5 -13
View File
@@ -56,19 +56,11 @@ module.exports = async kernel =>
method: 'shell.run',
params:
{
message: 'conda install conda-forge::cuda-runtime=12.4.1 cudnn=9.2.1.18 --yes',
conda:
{
path: path.resolve(__dirname, '.env')
}
}
},
{
when: '{{ gpu === "nvidia" }}',
method: 'shell.run',
params:
{
message: 'pip install tensorrt==10.4.0 --extra-index-url https://pypi.nvidia.com',
message:
[
'conda install conda-forge::cuda-runtime=12.4.1 cudnn=9.2.1.18 --yes',
'pip install tensorrt==10.4.0 --extra-index-url https://pypi.nvidia.com'
],
conda:
{
path: path.resolve(__dirname, '.env')