From e52967cd92f6537a9c14229fa0598f20c4e7194d Mon Sep 17 00:00:00 2001 From: Leonard Lausen Date: Mon, 19 Apr 2021 21:12:24 +0000 Subject: [PATCH] Update for MXNet 2 Alpha (#1546) --- README.md | 12 +++++------- docs/install/install-include.rst | 16 ++++++++-------- tools/docker/gluon_nlp_job.sh | 4 ++-- tools/docker/ubuntu18.04-cpu.Dockerfile | 2 +- tools/docker/ubuntu18.04-gpu.Dockerfile | 2 +- 5 files changed, 17 insertions(+), 19 deletions(-) diff --git a/README.md b/README.md index 7a7a293b95..8e2cedfa11 100644 --- a/README.md +++ b/README.md @@ -30,20 +30,18 @@ See our documents at https://nlp.gluon.ai/master/index.html. # Installation -First of all, install the latest MXNet. You may use the following commands: +First of all, install the MXNet 2 release such as MXNet 2 Alpha. You may use the +following commands: ```bash -# Install the version with CUDA 10.1 -python3 -m pip install -U --pre "mxnet-cu101>=2.0.0b20210121" -f https://dist.mxnet.io/python - # Install the version with CUDA 10.2 -python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20210121" -f https://dist.mxnet.io/python +python3 -m pip install -U --pre "mxnet-cu102>=2.0.0a" # Install the version with CUDA 11 -python3 -m pip install -U --pre "mxnet-cu110>=2.0.0b20210121" -f https://dist.mxnet.io/python +python3 -m pip install -U --pre "mxnet-cu110>=2.0.0a" # Install the cpu-only version -python3 -m pip install -U --pre "mxnet>=2.0.0b20210121" -f https://dist.mxnet.io/python +python3 -m pip install -U --pre "mxnet>=2.0.0a" ``` diff --git a/docs/install/install-include.rst b/docs/install/install-include.rst index 000d9ed056..ed67debe38 100644 --- a/docs/install/install-include.rst +++ b/docs/install/install-include.rst @@ -56,8 +56,8 @@ Select your preferences and run the install command. .. code-block:: bash - # Install Apache MXNet (incubating) (Disclaimer: This is an unreleased nightly build and is used for the purpose of development). - python3 -m pip install -U --pre "mxnet>=2.0.0" -f https://dist.mxnet.io/python + # Install Apache MXNet (incubating) 2 Alhpa or newer. + python3 -m pip install -U --pre "mxnet>=2.0.0a" # Install GluonNLP git clone https://github.com/dmlc/gluon-nlp.git @@ -68,10 +68,10 @@ Select your preferences and run the install command. .. code-block:: bash - # Install Apache MXNet (incubating) (Disclaimer: This is an unreleased nightly build and is used for the purpose of development). + # Install Apache MXNet (incubating) 2 Alhpa or newer. # Here we assume CUDA 10.2 is installed. You can change the number # according to your own CUDA version, e.g., cu101, cu110 - python3 -m pip install -U --pre "mxnet-cu102>=2.0.0" -f https://dist.mxnet.io/python + python3 -m pip install -U --pre "mxnet-cu102>=2.0.0a" # Install GluonNLP git clone https://github.com/dmlc/gluon-nlp.git @@ -84,8 +84,8 @@ Select your preferences and run the install command. .. code-block:: bash - # Install Apache MXNet (incubating) (Disclaimer: This is an unreleased nightly build and is used for the purpose of development). - python3 -m pip install -U --pre "mxnet>=2.0.0" -f https://dist.mxnet.io/python + # Install Apache MXNet (incubating) 2 Alhpa or newer. + python3 -m pip install -U --pre "mxnet>=2.0.0a" # Install GluonNLP git clone https://github.com/dmlc/gluon-nlp.git @@ -96,10 +96,10 @@ Select your preferences and run the install command. .. code-block:: bash - # Install Apache MXNet (incubating) (Disclaimer: This is an unreleased nightly build and is used for the purpose of development). + # Install Apache MXNet (incubating) 2 Alhpa or newer. # Here we assume CUDA 10.2 is installed. You can change the number # according to your own CUDA version, e.g., cu100, cu101 - python3 -m pip install -U --pre "mxnet-cu102>=2.0.0" -f https://dist.mxnet.io/python + python3 -m pip install -U --pre "mxnet-cu102>=2.0.0a" # Install GluonNLP git clone https://github.com/dmlc/gluon-nlp.git diff --git a/tools/docker/gluon_nlp_job.sh b/tools/docker/gluon_nlp_job.sh index 33cb318ccd..ba6209ec60 100755 --- a/tools/docker/gluon_nlp_job.sh +++ b/tools/docker/gluon_nlp_job.sh @@ -23,7 +23,7 @@ git fetch origin $SOURCE_REF:working git checkout working if [ $DEVICE == "cpu" ]; then - python3 -m pip install -U --quiet --pre "mxnet>=2.0.0b20210121" -f https://dist.mxnet.io/python --user + python3 -m pip install -U --quiet --pre "mxnet>=2.0.0b20210121" --user else # Due to the issue in https://forums.aws.amazon.com/thread.jspa?messageID=953912 # We need to manually configure the shm to ensure that Horovod is runnable. @@ -31,7 +31,7 @@ else umount shm mount -t tmpfs -o rw,nosuid,nodev,noexec,relatime,size=2G shm /dev/shm sudo python3 -m pip uninstall --quiet mxnet-cu102 - python3 -m pip install -U --quiet --pre "mxnet-cu102>=2.0.0b20210121" -f https://dist.mxnet.io/python/cu102 --user + python3 -m pip install -U --quiet --pre "mxnet-cu102>=2.0.0a" --user fi python3 -m pip install --quiet -e .[extras,dev] diff --git a/tools/docker/ubuntu18.04-cpu.Dockerfile b/tools/docker/ubuntu18.04-cpu.Dockerfile index 85a10b8c71..2f3e06d0fa 100644 --- a/tools/docker/ubuntu18.04-cpu.Dockerfile +++ b/tools/docker/ubuntu18.04-cpu.Dockerfile @@ -33,7 +33,7 @@ RUN bash /install/install_python_packages.sh RUN bash /install/install_tvm_cpu.sh # Install MXNet -RUN python3 -m pip install -U --pre "mxnet>=2.0.0b20210121" -f https://dist.mxnet.io/python --user +RUN python3 -m pip install -U --pre "mxnet>=2.0.0a" --user # Install PyTorch RUN python3 -m pip install "torch==1.7.1+cpu" torchvision -f https://download.pytorch.org/whl/torch_stable.html diff --git a/tools/docker/ubuntu18.04-gpu.Dockerfile b/tools/docker/ubuntu18.04-gpu.Dockerfile index be3f02053c..e4188f8a12 100644 --- a/tools/docker/ubuntu18.04-gpu.Dockerfile +++ b/tools/docker/ubuntu18.04-gpu.Dockerfile @@ -32,7 +32,7 @@ RUN bash /install/install_python_packages.sh RUN bash /install/install_tvm_gpu.sh # Install MXNet -RUN python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20210121" -f https://dist.mxnet.io/python --user +RUN python3 -m pip install -U --pre "mxnet-cu102>=2.0.0a" --user # Install PyTorch RUN python3 -m pip install "torch==1.8.1+cu102" torchvision -f https://download.pytorch.org/whl/torch_stable.html