onnx_inference.sh 1.8 KB
Newer Older
sunzhq2's avatar
sunzhq2 committed
1
2
3
4
5
export HIP_PRINTF_DEBUG_FOR_FP64=0
export HSA_FORCE_FINE_GRAIN_PCIE=1
# /datasets/cifar100/cifar-100-python/test
export HIP_VISIBLE_DEVICES=0

sunzhq2's avatar
sunzhq2 committed
6
numactl -N 0 -m 0 python resnet50_migraphx.py --model ./resnet50.onnx --dataset /datasets/cifar100/test --batch_size 24 --fp16 True
sunzhq2's avatar
sunzhq2 committed
7

sunzhq2's avatar
sunzhq2 committed
8
# nohup numactl -N 0 -m 0 python resnet50_migraphx.py --model ./resnet50.onnx --dataset ./datasets/cifar100/test --batch_size 24 --fp16 True 2>&1 | tee result_0.log &
sunzhq2's avatar
sunzhq2 committed
9

sunzhq2's avatar
sunzhq2 committed
10
11
# export HIP_VISIBLE_DEVICES=1
# nohup numactl -N 1 -m 1 python resnet50_migraphx.py --model ./resnet50.onnx --dataset ./datasets/cifar100/test --batch_size 24 --fp16 True 2>&1 | tee result_1.log &
sunzhq2's avatar
sunzhq2 committed
12

sunzhq2's avatar
sunzhq2 committed
13
14
# export HIP_VISIBLE_DEVICES=2
# nohup numactl -N 2 -m 2 python resnet50_migraphx.py --model ./resnet50.onnx --dataset ./datasets/cifar100/test --batch_size 24 --fp16 True 2>&1 | tee result_2.log &
sunzhq2's avatar
sunzhq2 committed
15

sunzhq2's avatar
sunzhq2 committed
16
17
# export HIP_VISIBLE_DEVICES=3
# nohup numactl -N 3 -m 3 python resnet50_migraphx.py --model ./resnet50.onnx --dataset ./datasets/cifar100/test --batch_size 24 --fp16 True 2>&1 | tee result_3.log &
sunzhq2's avatar
sunzhq2 committed
18
19
20
21
22
23
24
25
26
27
28
29

# export HIP_VISIBLE_DEVICES=4
# nohup python resnet50_migraphx.py --model /workspace/mmpretrain-main/resnet50.onnx --dataset /datasets/cifar100/ --batch_size 24 --fp16 True 2>&1 | tee result_4.log &

# export HIP_VISIBLE_DEVICES=5
# nohup python resnet50_migraphx.py --model /workspace/mmpretrain-main/resnet50.onnx --dataset /datasets/cifar100/ --batch_size 24 --fp16 True 2>&1 | tee result_5.log &

# export HIP_VISIBLE_DEVICES=6
# nohup python resnet50_migraphx.py --model /workspace/mmpretrain-main/resnet50.onnx --dataset /datasets/cifar100/ --batch_size 24 --fp16 True 2>&1 | tee result_6.log &

# export HIP_VISIBLE_DEVICES=7
# nohup python resnet50_migraphx.py --model /workspace/mmpretrain-main/resnet50.onnx --dataset /datasets/cifar100/ --batch_size 24 --fp16 True 2>&1 | tee result_7.log &