remove pyrsmi and nvgpu deps

nod-ai · Nov 26, 2024 · 0a4d628 · 0a4d628
1 parent 61da32e
commit 0a4d628
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 14 deletions.
diff --git a/requirements/common_gpu.txt b/requirements/common_gpu.txt
diff --git a/requirements/common_rocm.txt b/requirements/common_rocm.txt
diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py
@@ -167,16 +167,6 @@ def install_python_packages(
         # developer.txt also installs packages from common.txt
         os.system(f"{sys.executable} -m pip install -U -r {requirements_file_path}")
 
-        # Install dependencies for NVIDIA GPU
-        if not isinstance(cuda_version, type(None)):
-            gpu_requirements_file = os.path.join("requirements", "common_gpu.txt")
-            os.system(f"{sys.executable} -m pip install -U -r {gpu_requirements_file}")
-
-        # Install dependencies for AMD GPU
-        if not isinstance(rocm_version, type(None)):
-            gpu_requirements_file = os.path.join("requirements", "common_rocm.txt")
-            os.system(f"{sys.executable} -m pip install -U -r {gpu_requirements_file}")
-
         # Install dependencies for Inferentia2
         if args.neuronx:
             neuronx_requirements_file = os.path.join("requirements", "neuronx.txt")

diff --git a/ts_scripts/print_env_info.py b/ts_scripts/print_env_info.py
@@ -112,7 +112,7 @@ def run_with_pip(pip):
         elif package_name == "torch":
             grep_cmd = 'grep "' + package_name + '"'
         else:
-            grep_cmd = r'grep "numpy\|pytest\|pylint\|transformers\|psutil\|wheel\|requests\|sentencepiece\|pillow\|captum\|nvgpu\|pygit2\|torch"'
+            grep_cmd = r'grep "numpy\|pytest\|pylint\|transformers\|psutil\|wheel\|requests\|sentencepiece\|pillow\|captum\|pygit2\|torch"'
         return run_and_read_all(pip + " list --format=freeze | " + grep_cmd)
 
     out = run_with_pip("pip3")