remove: torchvision dependency when log torch model. fixed #4381 (#5153)

* remove: torchvision dependency when log torch model. fixed #4381 Signed-off-by: zhouweipeng <zhouweipeng@lattebank.com> * docs: rm torchvision Signed-off-by: zhouweipeng <zhouweipeng@lattebank.com> Co-authored-by: zhouweipeng <zhouweipeng@lattebank.com>
mlflow · Dec 20, 2021 · 549575d · 549575d
1 parent 5fa575e
commit 549575d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/mlflow/pytorch/__init__.py b/mlflow/pytorch/__init__.py
@@ -68,7 +68,6 @@ def get_default_pip_requirements():
             _get_pinned_requirement,
             [
                 "torch",
-                "torchvision",
                 # We include CloudPickle in the default environment because
                 # it's required by the default pickle module used by `save_model()`
                 # and `log_model()`: `mlflow.pytorch.pickle_module`.
@@ -103,7 +102,6 @@ def get_default_conda_env():
                    'channels': ['conda-forge'],
                    'dependencies': ['python=3.7.5',
                                     {'pip': ['torch==1.5.1',
-                                             'torchvision==0.6.1',
                                              'mlflow',
                                              'cloudpickle==1.6.0']}]}
     """