Fix OpenBLAS warning (#583)
Fixes the error msg: "OpenBLAS warning: precompiled NUM_THREADS exceeded, adding auxiliary array for thread metadata"
This commit is contained in:
parent
413c0d285d
commit
cdbb76cb6f
@ -10,6 +10,7 @@ import warnings
|
||||
import numpy as np
|
||||
import torch
|
||||
|
||||
os.environ['OPENBLAS_NUM_THREADS'] = '1'
|
||||
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
|
||||
import logging
|
||||
import threading
|
||||
|
Loading…
x
Reference in New Issue
Block a user