File size: 133 Bytes
d1b91e7
 
 
 
 
1
2
3
4
5
6
import os

os.environ["OMP_NUM_THREADS"] = "1"
os.environ['TF_NUM_INTEROP_THREADS'] = '1'
os.environ['TF_NUM_INTRAOP_THREADS'] = '1'