From e940837ee80447ee97fb65ac8390314bf446bf24 Mon Sep 17 00:00:00 2001 From: Stella Laurenzo Date: Mon, 18 Mar 2024 14:06:28 -0700 Subject: [PATCH] Try rocm6 --- dockerfiles/amdgpu_ubuntu_jammy_x86_64.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dockerfiles/amdgpu_ubuntu_jammy_x86_64.Dockerfile b/dockerfiles/amdgpu_ubuntu_jammy_x86_64.Dockerfile index 102afdf..524667c 100644 --- a/dockerfiles/amdgpu_ubuntu_jammy_x86_64.Dockerfile +++ b/dockerfiles/amdgpu_ubuntu_jammy_x86_64.Dockerfile @@ -23,7 +23,7 @@ RUN rmdir /install-amdvlk # Install the ROCM driver. # In order to use this in the container, you must pass /dev/dri and /dev/kfd in as: # docker run --device=/dev/kfd --device=/dev/dri rocminfo -ARG ROCM_VERSION=5.6 +ARG ROCM_VERSION=6.0 WORKDIR /install-rocm RUN mkdir --parents --mode=0755 /etc/apt/keyrings && wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \ gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null && \