From 0a4d628ca4c8a446efb7749a07d94b9ea8e45c38 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rony=20Lepp=C3=A4nen?= Date: Tue, 26 Nov 2024 13:09:18 +0000 Subject: [PATCH] remove pyrsmi and nvgpu deps --- requirements/common_gpu.txt | 2 -- requirements/common_rocm.txt | 1 - ts_scripts/install_dependencies.py | 10 ---------- ts_scripts/print_env_info.py | 2 +- 4 files changed, 1 insertion(+), 14 deletions(-) delete mode 100644 requirements/common_gpu.txt delete mode 100644 requirements/common_rocm.txt diff --git a/requirements/common_gpu.txt b/requirements/common_gpu.txt deleted file mode 100644 index 1e893cc7c1..0000000000 --- a/requirements/common_gpu.txt +++ /dev/null @@ -1,2 +0,0 @@ -nvgpu; sys_platform != 'win32' -nvgpu==0.10.0; sys_platform == 'win32' diff --git a/requirements/common_rocm.txt b/requirements/common_rocm.txt deleted file mode 100644 index 20789dd473..0000000000 --- a/requirements/common_rocm.txt +++ /dev/null @@ -1 +0,0 @@ -pyrsmi; sys_platform == 'linux' \ No newline at end of file diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py index e66b5b5701..4d464e03fe 100644 --- 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 index 430bb98e29..b6d194b688 100644 --- 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")