diff --git a/torchtext/__init__.py b/torchtext/__init__.py index 015b51501f..28f8f21c23 100644 --- a/torchtext/__init__.py +++ b/torchtext/__init__.py @@ -1,14 +1,13 @@ import os +from torch.hub import _get_torch_home + # the following import has to happen first in order to load the torchtext C++ library from torchtext import _extension # noqa: F401 _TEXT_BUCKET = "https://download.pytorch.org/models/text/" -_TORCH_HOME = os.getenv("TORCH_HOME") -if _TORCH_HOME is None: - _TORCH_HOME = "~/.cache/torch" # default -_CACHE_DIR = os.path.expanduser(os.path.join(_TORCH_HOME, "text")) +_CACHE_DIR = os.path.expanduser(os.path.join(_get_torch_home(), "text")) from . import data, datasets, experimental, functional, models, nn, transforms, utils, vocab