Skip to content

Commit b6f45fc

Browse files
authored
Update sample.json
updated with requirements packages
1 parent c0b4af3 commit b6f45fc

File tree

1 file changed

+4
-3
lines changed
  • AI-and-Analytics/Features-and-Functionality/IntelPyTorch_GPU_InferenceOptimization_with_AMP

1 file changed

+4
-3
lines changed

AI-and-Analytics/Features-and-Functionality/IntelPyTorch_GPU_InferenceOptimization_with_AMP/sample.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{
22
"guid": "91B5893E-028B-47BC-9D84-395A60CC1B33",
3-
"name": "Intel PyTorch GPU with AMP",
3+
"name": "Intel PyTorch GPU Inference Optimization with AMP",
44
"categories": ["Toolkit/oneAPI AI And Analytics/AI Features and Functionality"],
55
"description": "This sample illustrates how to use AMP BFLOAT16 in PyTorch on Intel dGPU",
66
"builder": ["cli"],
@@ -10,10 +10,11 @@
1010
"ciTests": {
1111
"linux": [
1212
{
13-
"id": "intel pytorch gpu with amp",
13+
"id": "intel pytorch gpu inference optimization with amp",
1414
"steps": [
1515
"source activate pytorch-gpu",
16-
"python IntelPyTorch_GPU_with_AMP.py",
16+
"pip install -r requirements.txt",
17+
"python IntelPyTorch_GPU_InferenceOptimization_with_AMP.py",
1718
"/opt/intel/oneapi/intelpython/latest/envs/pytorch/bin/python -m ipykernel install --user --name=pytorch-gpu",
1819
"python ci_test.py"
1920
]

0 commit comments

Comments
 (0)