mirror of https://github.com/hpcaitech/ColossalAI
10 lines
334 B
Python
10 lines
334 B
Python
try:
|
|
from ._meta_registrations import *
|
|
except:
|
|
import torch
|
|
print(f'_meta_registrations seems to be incompatible with PyTorch {torch.__version__}.')
|
|
from .initialize import (initialize, launch, launch_from_openmpi, launch_from_slurm, launch_from_torch,
|
|
get_default_parser)
|
|
|
|
__version__ = '0.1.9'
|