Commit faa462fa authored by Evelina Bakhturina's avatar Evelina Bakhturina
Browse files

update 



Signed-off-by: default avatarEvelina Bakhturina <ebakhturina@nvidia.com>
parent 5f871088

README.md

+1 −1
Original line number Diff line number Diff line
@@ -36,7 +36,7 @@ We officially support only python 3.6, pytorch 1.5, cuda 10, and nccl 2.6 versio 


To use this repo please install the latest supported versions of PyTorch with GPU support and NVIDIA [APEX](https://github.com/NVIDIA/apex#quick-start). We strongly recommend using one of [NGC's recent PyTorch containers](https://ngc.nvidia.com/catalog/containers/nvidia:pytorch) (the latest compatible version at time of publication can be pulled with `docker pull nvcr.io/nvidia/pytorch:20.03-py3`). Data preprocessing requires [NLTK](https://www.nltk.org/install.html), though this is not required for training, evaluation or downstream tasks.



To use megatron you can either clone the repo or install it via pip:

To use megatron you can either clone the repo or install it via pip (make sure python3-dev is installed):

<pre>

pip install megatron-lm

</pre>

setup.py

+0 −2
Original line number Diff line number Diff line
@@ -71,8 +71,6 @@ setuptools.setup( 
        # Indicate what your project relates to

        'Topic :: Scientific/Engineering :: Artificial Intelligence',

        'Topic :: Software Development :: Libraries :: Python Modules',

        # Pick your license as you wish (should match "license" above)

        'License :: OSI Approved :: Apache Software License',

        # Supported python versions

        'Programming Language :: Python :: 3.6',

        'Programming Language :: Python :: 3.7',