123456789101112131415161718192021222324252627282930313233343536 |
- from pathlib import Path
- import os
- cwd = Path(".").resolve()
- print("running 'check_installation.py' from:", cwd)
- # Old versions of numpy/torch can prevent loading the .so files
- import torch
- print("torch:", torch.__version__)
- import numpy
- print("numpy:", numpy.__version__)
- import fairseq
- print("Fairseq installed at:", fairseq.__file__)
- import fairseq.criterions
- import fairseq.dataclass.configs
- import _imp
- print("Should load following .so suffixes:", _imp.extension_suffixes())
- so_files = list(Path(fairseq.__file__).parent.glob("*.so"))
- so_files.extend(Path(fairseq.__file__).parent.glob("data/*.so"))
- print("Found following .so files:")
- for so_file in so_files:
- print(f"- {so_file}")
- from fairseq import libbleu
- print("Found libbleu at", libbleu.__file__)
- from fairseq.data import data_utils_fast
- print("Found data_utils_fast at", data_utils_fast.__file__)
|