Update beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Cla…

…ssification_Resnet_152_Tesla_T4_GPU.txt (#31341) Fix #30644
apache · May 18, 2024 · 71f2278 · 71f2278
1 parent f851cbb
commit 71f2278
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ce_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt b/...ce_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
@@ -29,8 +29,8 @@
 --influx_measurement=torch_inference_imagenet_resnet152_tesla_t4
 --pretrained_model_name=resnet152
 --device=GPU
---experiments=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver
+--experiments=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver:5xx
 --sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
 --input_file=gs://apache-beam-ml/testing/inputs/openimage_50k_benchmark.txt
 --model_state_dict_path=gs://apache-beam-ml/models/torchvision.models.resnet152.pth
---runner=DataflowRunner
+--runner=DataflowRunner