diff --git a/setup.py b/setup.py index 8e41548..4683545 100644 --- a/setup.py +++ b/setup.py @@ -18,5 +18,5 @@ classifiers=["Development Status :: 2 - Pre-Alpha"], keywords="deep learning, distributed computing, parallel computing", packages=find_packages(), - install_requires=["torch", "mpi4py"], + install_requires=["torch"], )