From bb6b3e2e3f3c7440dfb634610a84523fd51146da Mon Sep 17 00:00:00 2001
From: evshiron <evshiron@gmail.com>
Date: Mon, 14 Aug 2023 00:57:40 +0800
Subject: [PATCH] fix: fix tensorflow installer for navi 3x

---
 installer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/installer.py b/installer.py
index b9241cb0c..05bbeb5bb 100644
--- a/installer.py
+++ b/installer.py
@@ -344,7 +344,8 @@ def check_torch():
             if arch == 'navi3x':
                 os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '11.0.0')
                 # do not use tensorflow-rocm for navi 3x
-                os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.13.0')
+                if os.environ.get('TENSORFLOW_PACKAGE') == 'tensorflow-rocm':
+                    os.environ['TENSORFLOW_PACKAGE'] = 'tensorflow==2.13.0'
             elif arch == 'navi2x':
                 os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '10.3.0')
             else: