Skip to content

Commit

Permalink
update npu python version in tipc config (#640)
Browse files Browse the repository at this point in the history
  • Loading branch information
YanhuiDua authored Jul 25, 2023
1 parent c50f176 commit 3f9eb13
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions test_tipc/test_train_inference_python_npu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,8 @@ fi
sed -i "s/use_gpu/use_npu/g" $FILENAME
# disable benchmark as AutoLog required nvidia-smi command
sed -i "s/--enable_benchmark:True/--enable_benchmark:False/g" $FILENAME
# python has been updated to version 3.9 for npu backend
sed -i "s/python3.7/python3.9/g" $FILENAME
dataline=`cat $FILENAME`

# change gpu to npu in execution script
Expand Down

0 comments on commit 3f9eb13

Please sign in to comment.