diff --git a/requirements/optional.txt b/requirements/optional.txt
index ca3c4e77..5cf26773 100644
--- a/requirements/optional.txt
+++ b/requirements/optional.txt
@@ -1,2 +1,2 @@
-albumentations>=0.3.2 --no-binary imgaug,albumentations
+albumentations>=0.3.2
 requests
diff --git a/setup.py b/setup.py
index 3f9368a2..0ed33fb5 100644
--- a/setup.py
+++ b/setup.py
@@ -163,10 +163,7 @@ if __name__ == '__main__':
         description='OpenMMLab Image Classification Toolbox and Benchmark',
         long_description=readme(),
         long_description_content_type='text/markdown',
-        author='MMClassification Contributors',
-        author_email='openmmlab@gmail.com',
         keywords='computer vision, image classification',
-        url='https://github.com/open-mmlab/mmclassification',
         packages=find_packages(exclude=('configs', 'tools', 'demo')),
         include_package_data=True,
         classifiers=[
@@ -179,8 +176,16 @@ if __name__ == '__main__':
             'Programming Language :: Python :: 3.7',
             'Programming Language :: Python :: 3.8',
             'Programming Language :: Python :: 3.9',
+            'Topic :: Scientific/Engineering :: Artificial Intelligence',
         ],
+        url='https://github.com/open-mmlab/mmclassification',
+        author='MMClassification Contributors',
+        author_email='openmmlab@gmail.com',
         license='Apache License 2.0',
-        tests_require=parse_requirements('requirements/tests.txt'),
         install_requires=parse_requirements('requirements/runtime.txt'),
+        extras_require={
+            'all': parse_requirements('requirements.txt'),
+            'tests': parse_requirements('requirements/tests.txt'),
+            'optional': parse_requirements('requirements/optional.txt'),
+        },
         zip_safe=False)