From ceb388e8a08561c2281924e37f8c97bfd5ac2a80 Mon Sep 17 00:00:00 2001 From: kingbri Date: Tue, 2 Jan 2024 21:00:22 -0500 Subject: [PATCH] Start: Override ROCm env variables These are used for supporting GPUs that are not on the "officially supported list". Signed-off-by: kingbri --- start.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/start.py b/start.py index 8fe34fa..c4b9334 100644 --- a/start.py +++ b/start.py @@ -15,6 +15,11 @@ def get_requirements_file(): # TODO: Check if the user has an AMD gpu on windows if ROCM_PATH: requirements_name = "requirements-amd" + + # Also override env vars for ROCm support on non-supported GPUs + os.environ["ROCM_PATH"] = "/opt/rocm" + os.environ["HSA_OVERRIDE_GFX_VERSION"] = "10.3.0" + os.environ["HCC_AMDGPU_TARGET"] = "gfx1030" elif CUDA_PATH: cuda_version = pathlib.Path(CUDA_PATH).name if "12" in cuda_version: