Hotfix
This commit is contained in:
parent
e6cc97daab
commit
30713c3c0e
@ -3,7 +3,7 @@
|
||||
import os
|
||||
import sys
|
||||
# single thread doubles performance of gpu-mode - needs to be set before torch import
|
||||
if any(arg.startswith('--gpu-vendor=') for arg in sys.argv):
|
||||
if any(arg.startswith('--gpu-vendor') for arg in sys.argv):
|
||||
os.environ['OMP_NUM_THREADS'] = '1'
|
||||
import platform
|
||||
import signal
|
||||
|
Loading…
Reference in New Issue
Block a user