add MKL ENV
Browse files
script.py
CHANGED
|
@@ -105,6 +105,9 @@ import subprocess
|
|
| 105 |
import sys
|
| 106 |
import os
|
| 107 |
|
|
|
|
|
|
|
|
|
|
| 108 |
def download_packages(packages, folder='packages/torch'):
|
| 109 |
"""
|
| 110 |
Downloads packages as .whl files into the specified folder using pip.
|
|
|
|
| 105 |
import sys
|
| 106 |
import os
|
| 107 |
|
| 108 |
+
os.environ['MKL_THREADING_LAYER'] = 'GNU'
|
| 109 |
+
os.environ['MKL_SERVICE_FORCE_INTEL'] = '1'
|
| 110 |
+
|
| 111 |
def download_packages(packages, folder='packages/torch'):
|
| 112 |
"""
|
| 113 |
Downloads packages as .whl files into the specified folder using pip.
|