import torch

from setuptools import setup, find_packages

if not torch.cuda.is_available():

print("Warning: Torch did not find available GPUs on this system.\n",

"If your intention is to cross-compile, this is not an error.")

print("torch.__version__ = ", torch.__version__)

TORCH_MAJOR = int(torch.__version__.split('.')[0])

TORCH_MINOR = int(torch.__version__.split('.')[1])

if TORCH_MAJOR == 0 and TORCH_MINOR < 4:

raise RuntimeError("APEx requires Pytorch 0.4 or newer.\n" +

"The latest stable release can be obtained from https://pytorch.org/")

setup(

name='apex',

version='0.1',

packages=find_packages(exclude=('build',

'csrc',

'include',

'tests',

'dist',

'docs',

'tests',

'examples',

'apex.egg-info',)),

description='PyTorch Extensions written by NVIDIA',

)