|
"""Root package info.""" |
|
|
|
import logging |
|
import os |
|
import sys |
|
|
|
from lightning_utilities.core.imports import package_available |
|
|
|
if os.path.isfile(os.path.join(os.path.dirname(__file__), "__about__.py")): |
|
from lightning_fabric.__about__ import * |
|
if os.path.isfile(os.path.join(os.path.dirname(__file__), "__version__.py")): |
|
from lightning_fabric.__version__ import version as __version__ |
|
elif package_available("lightning"): |
|
from lightning_fabric import __version__ |
|
|
|
_root_logger = logging.getLogger() |
|
_logger = logging.getLogger(__name__) |
|
_logger.setLevel(logging.INFO) |
|
|
|
if not _root_logger.hasHandlers(): |
|
_logger.addHandler(logging.StreamHandler()) |
|
_logger.propagate = False |
|
|
|
|
|
|
|
|
|
|
|
os.environ["PYTORCH_NVML_BASED_CUDA_CHECK"] = "1" |
|
|
|
|
|
if sys.platform == "win32": |
|
os.environ["USE_LIBUV"] = "0" |
|
|
|
|
|
from lightning_fabric.fabric import Fabric |
|
from lightning_fabric.utilities.seed import seed_everything |
|
from lightning_fabric.utilities.warnings import disable_possible_user_warnings |
|
from lightning_fabric.wrappers import is_wrapped |
|
|
|
|
|
import lightning_fabric._graveyard |
|
|
|
__all__ = ["Fabric", "seed_everything", "is_wrapped"] |
|
|
|
|
|
if os.environ.get("POSSIBLE_USER_WARNINGS", "").lower() in ("0", "off"): |
|
disable_possible_user_warnings() |
|
|