update dtk version

python/setup.py

@@ -18,6 +18,13 @@ class BinaryDistribution(Distribution):
 
 CURRENT_DIR = os.path.dirname(__file__)
 
+def get_dtk_version():
+    ROCM_PATH = os.getenv('ROCM_PATH', "/opt/dtk/")
+    dtk_path = ROCM_PATH + '/.info/rocm_version'
+    with open(dtk_path, 'r') as file:
+        content = file.read().strip()
+    dtk_version = "dtk"+content.replace('.', '')
+    return dtk_version
 
 def get_lib_path():
     """Get library path, name and version"""
@@ -234,10 +241,9 @@ install_requires = [
 ]
 if "DGLBACKEND" in os.environ and os.environ["DGLBACKEND"] != "pytorch":
     install_requires.pop(install_requires.index("torchdata>=0.5.0"))
-
 setup(
     name="dgl" + os.getenv("DGL_PACKAGE_SUFFIX", ""),
-    version=VERSION+str('+das.opt1.dtk2504'),
+    version=VERSION+str('+das.opt1.')+get_dtk_version(),
     description="Deep Graph Library",
     zip_safe=False,
     maintainer="DGL Team",