colin1842
/

baseline-cuda121

Model card Files Files and versions

colin1842 commited on Jun 7, 2024

Commit

6e3d78e

·

1 Parent(s): 6e29b1f

MKL

Files changed (1) hide show

script.py +2 -2

script.py CHANGED Viewed

@@ -106,8 +106,8 @@ import sys
 import os
 import numpy as np
-# os.environ['MKL_THREADING_LAYER'] = 'GNU'
-# os.environ['MKL_SERVICE_FORCE_INTEL'] = '1'
 # def download_packages(packages, folder='packages/torch'):
 # 	"""

 import os
 import numpy as np
+os.environ['MKL_THREADING_LAYER'] = 'GNU'
+os.environ['MKL_SERVICE_FORCE_INTEL'] = '1'
 # def download_packages(packages, folder='packages/torch'):
 # 	"""