Skip to content

GitLab

"test/gtest-1.10.0/googlemock/scripts/generator/cpp/ast.py" did not exist on "e2818c423e5058a02f46ce2e519a82742a8ccac9"
Commit 451ad165 authored by yangzhong's avatar yangzhong
Browse files

update README

parent 84b2558d
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
README.md
View file @ 451ad165
......@@ -7,7 +7,12 @@
单卡即可推理
测试方法:
# 单卡推理
export HIP_VISIBLE_DEVICES=0 # 指定卡号，不指定时默认使用第一张卡
python infer_resnet50.py
# 8卡推理
self.batch_size = 128 * 8 # 修改卡数
export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python infer_resnet50_8.py
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment