From 7eee0725a5496b0e7991ce6f04fff608c7a7301d Mon Sep 17 00:00:00 2001
From: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com>
Date: Mon, 21 Oct 2024 09:01:26 +0530
Subject: [PATCH] updations for amd - docs

---
 main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index a4f09d559..8b6ce3ea5 100755
--- a/main.py
+++ b/main.py
@@ -44,8 +44,9 @@ def mlperf_inference_implementation_readme(spaces, model, implementation, *, imp
             frameworks = [ "TensorRT" ]
         
         elif implementation == "amd":
-            devices = [ "ROCm" ]
+            devices = [ "cuda" ]
             frameworks = [ "pytorch" ]
+            execution_envs.remove("Docker")
         
         elif implementation == "neuralmagic":
             devices = [ "CUDA" ]