Spaces:

tsi-org
/

FaceswapperGPU

Build error

tsi-org commited on Oct 28, 2023

Commit

9f1c8f8

1 Parent(s): 161514d

Update roop/core.py

Files changed (1) hide show

roop/core.py CHANGED Viewed

@@ -2,7 +2,7 @@
 import os
 import sys
-# os.environ["CUDA_VISIBLE_DEVICES"] = ""
 # single thread doubles cuda performance - needs to be set before torch import
 if any(arg.startswith('--execution-provider') for arg in sys.argv):
     os.environ['OMP_NUM_THREADS'] = '1'

 import os
 import sys
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 # single thread doubles cuda performance - needs to be set before torch import
 if any(arg.startswith('--execution-provider') for arg in sys.argv):
     os.environ['OMP_NUM_THREADS'] = '1'