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" ]