From 536ee9c9821002732830318b6d28bab4b6513073 Mon Sep 17 00:00:00 2001 From: rraminen Date: Tue, 30 Nov 2021 21:34:58 +0000 Subject: [PATCH] requirements-rocm.txt is not required anymore --- requirements/requirements-rocm.txt | 7 ------- setup.py | 3 --- 2 files changed, 10 deletions(-) delete mode 100644 requirements/requirements-rocm.txt diff --git a/requirements/requirements-rocm.txt b/requirements/requirements-rocm.txt deleted file mode 100644 index e0e1d0c482ce..000000000000 --- a/requirements/requirements-rocm.txt +++ /dev/null @@ -1,7 +0,0 @@ -#torch>=1.2 -#torchvision>=0.4.0 -tqdm -tensorboardX==1.8 -ninja -numpy -psutil diff --git a/setup.py b/setup.py index 6694b623cbea..75bbbe342945 100755 --- a/setup.py +++ b/setup.py @@ -49,9 +49,6 @@ def fetch_requirements(path): install_requires = fetch_requirements('requirements/requirements.txt') -if is_rocm_pytorch: - print("NOTE: Please manually install torch and torchvision packages for ROCm") - install_requires = fetch_requirements('requirements/requirements-rocm.txt') extras_require = { '1bit_mpi' : fetch_requirements('requirements/requirements-1bit-mpi.txt'), '1bit': [], # Will add proper cupy version below