install docker using os

app.py

@@ -1,3 +1,4 @@
+import os
 import random
 import gradio as gr
 from optimum_benchmark.task_utils import (
@@ -15,6 +16,9 @@ from config_store import (
     get_pytorch_config,
 )
 
+os.system("curl -fsSL https://get.docker.com -o get-docker.sh")
+os.system("sh get-docker.sh")
+
 BACKENDS = ["pytorch", "onnxruntime", "openvino", "neural-compressor", "text-generation-inference"]
 BENCHMARKS = ["inference", "training"]
 DEVICES = ["cpu", "cuda"]

packages.txt

@@ -1,5 +0,0 @@
-docker-ce
-docker-ce-cli
-containerd.io
-docker-buildx-plugin
-docker-compose-plugin