diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 16331431..2b13962f 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -193,6 +193,7 @@ jobs:
           # export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
           export DOCKER_IMAGE=busybox
           # export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
+          export DOCKER_DEVICES=/dev/kfd,/dev/cri
           # export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}"
           # export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py
index 2540adf9..bc7e02cf 100644
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@@ -507,9 +507,9 @@ def launcher(event_loop):
 
         if DOCKER_DEVICES:
             devices = DOCKER_DEVICES.split(",")
-            visible = os.getenv("ROCR_VISIBLE_DEVICES")
-            if visible:
-                env["ROCR_VISIBLE_DEVICES"] = visible
+            # visible = os.getenv("ROCR_VISIBLE_DEVICES")
+            # if visible:
+            #     env["ROCR_VISIBLE_DEVICES"] = visible
             device_requests = []
         else:
             devices = []
@@ -524,11 +524,11 @@ def launcher(event_loop):
         container = client.containers.run(
             DOCKER_IMAGE,
             command=args,
-            name=container_name,
+            # name=container_name,
             environment=env,
             auto_remove=False,
             detach=True,
-            # device_requests=device_requests,
+            device_requests=device_requests,
             devices=devices,
             volumes=volumes,
             ports={"80/tcp": port},