Skip to content

Commit e2249f5

Browse files
authored
update sample.json to test for both CPU & GPU (#1979)
1 parent 8189b0f commit e2249f5

2 files changed

Lines changed: 16 additions & 13 deletions

File tree

  • AI-and-Analytics/Getting-Started-Samples/Intel_Extension_For_TensorFlow_GettingStarted

AI-and-Analytics/Getting-Started-Samples/Intel_Extension_For_TensorFlow_GettingStarted/sample.json

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,20 +13,23 @@
1313
"ciTests": {
1414
"linux": [{
1515
"env": [
16-
"source /opt/intel/oneapi/setvars.sh --force",
17-
"conda create --name user-tensorflow-gpu --clone tensorflow-gpu",
18-
"conda activate user-tensorflow-gpu",
19-
"pip install -r requirements.txt",
20-
"~/.conda/envs/user-tensorflow-gpu/bin/python -m ipykernel install --user --name=user-tensorflow-gpu",
21-
"conda deactivate",
22-
"conda create --name user-tensorflow --clone tensorflow",
23-
"conda activate user-tensorflow",
24-
"pip install -r requirements.txt",
25-
"~/.conda/envs/user-tensorflow/bin/python -m ipykernel install --user --name=user-tensorflow ",
26-
"conda deactivate"
27-
],
16+
"source /opt/intel/oneapi/setvars.sh --force",
17+
"conda create --name user-tensorflow --clone tensorflow",
18+
"conda activate user-tensorflow",
19+
"pip install -r requirements.txt",
20+
"pip install nbformat nbconvert",
21+
"python -m ipykernel install --user --name=user-tensorflow",
22+
"conda deactivate",
23+
"conda create --name user-tensorflow-gpu --clone tensorflow-gpu",
24+
"conda activate user-tensorflow-gpu",
25+
"pip install -r requirements.txt",
26+
"pip install nbformat nbconvert",
27+
"python -m ipykernel install --user --name=user-tensorflow-gpu",
28+
"conda deactivate"
29+
],
2830
"id": "itex_sample_test",
2931
"steps": [
32+
"conda activate user-tensorflow",
3033
"python scripts/ci_test.py"
3134
]
3235
}]

AI-and-Analytics/Getting-Started-Samples/Intel_Extension_For_TensorFlow_GettingStarted/scripts/ci_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,4 +22,4 @@ def runJupyterNotebook(input_notebook_filename, output_notebook_filename, conda_
2222

2323
runJupyterNotebook(os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))),
2424
'ResNet50_Inference.ipynb'),
25-
'ResNet50_Inference_result.ipynb', 'user-tensorflow-gpu')
25+
'ResNet50_Inference_result.ipynb', 'user-tensorflow')

0 commit comments

Comments
 (0)