Start: Override ROCm env variables
These are used for supporting GPUs that are not on the "officially supported list". Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
parent
c980f35e1b
commit
ceb388e8a0
1 changed files with 5 additions and 0 deletions
5
start.py
5
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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue